Home / Machine Learning / JAX JAX JAX is Autograd and XLA, brought together for high-performance machine learning research. Package 35.2k stars GitHub Back to Machine Learning
