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
-
Will jax support AMD and ROCm in the future?
Beta Was this translation helpful? Give feedback.
All reactions