Jax in embedded GPUs #4503
Jax in embedded GPUs
#4503
-
|
Hello Has anyone tried using JAX with embedded GPUs? Like the Jetson Xavier? Thank you |
Beta Was this translation helpful? Give feedback.
Answered by
jekbradbury
Oct 11, 2020
Replies: 1 comment
-
|
If you can successfully build a CUDA jaxlib on a Jetson board, I think we expect JAX would run fine. I don't think anyone's done it before, though. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
aizzaac
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
If you can successfully build a CUDA jaxlib on a Jetson board, I think we expect JAX would run fine. I don't think anyone's done it before, though.