JAX on CPU cluster #24496
Unanswered
xlaLearner
asked this question in
Q&A
JAX on CPU cluster
#24496
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
After reading the “Introduction to parallel programming” in the official documentation of JAX, I have a question, which is: How about parallel performance on CPU clusters when JAX has good parallel performance on multiple GPUs and TPUs? In CPU cluster based computing, is the distributed model of JAX based on a single CPU or on cores within CPUs working in parallel?
Beta Was this translation helpful? Give feedback.
All reactions