Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Python][Docs] Mention Arrow->JAX conversion via dlpack in the examples #44229

Closed
felipecrv opened this issue Sep 25, 2024 · 1 comment
Closed

Comments

@felipecrv
Copy link
Contributor

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

felipecrv added a commit that referenced this issue Sep 26, 2024
### 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>
@felipecrv felipecrv added this to the 18.0.0 milestone Sep 26, 2024
@felipecrv
Copy link
Contributor Author

Issue resolved by pull request 44230
#44230

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

1 participant