You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
JAX is a popular numeric computing library. People googling for how to convert pyarrow arrays to JAX arrays might not infer by themselves that it can be done via the dlpack protocol.
Component(s)
Python
The text was updated successfully, but these errors were encountered:
### Rationale for this change
Explicitly mention in the docs a way that PyArrow can interop with [JAX](https://github.com/jax-ml/jax).
### What changes are included in this PR?
- Tweaks to the phrasing
- Two JAX examples: one for `jax.numpy` and another for `jax.dlpack`
### Are these changes tested?
N/A
* GitHub Issue: #44229
Authored-by: Felipe Oliveira Carvalho <felipekde@gmail.com>
Signed-off-by: Felipe Oliveira Carvalho <felipekde@gmail.com>
Describe the enhancement requested
JAX is a popular numeric computing library. People googling for how to convert pyarrow arrays to JAX arrays might not infer by themselves that it can be done via the dlpack protocol.
Component(s)
Python
The text was updated successfully, but these errors were encountered: