ROCm or AMD GPUs support
#6489
Replies: 1 comment 1 reply
-
|
JAX actually supports AMD and ROCm right now, although it's community supported and you need to build |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
Does jax currently support nvlink? |
Beta Was this translation helpful? Give feedback.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Will jax support AMD and ROCm in the future?
Beta Was this translation helpful? Give feedback.
All reactions