felipecrv commented on PR #44230:
URL: https://github.com/apache/arrow/pull/44230#issuecomment-2375491612
> One question, `jax.numpy.from_dlpack` ("Returns a Array representation of
a DLPack tensor") and `jax.dlpack.from_dlpack` ("Construct a JAX array via
DLPack") seem like redundant APIs, do JAX users tend to understand which is
which or could we make it clear why a user would pick one or other in this
example?
Yes. JAX itself is a nice to use Python library wrapping the XLA compiler.
Then it provides public interfaces inspired by famous Python libraries: there
is `jax.numpy` and `jax.scipy` for instance. `jax.dlpack` is the standalone
entrypoint to `dlpack` code in case you are not using JAX via the familiar
interfaces.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
