Home / Deep Learning / JAX JAX JAX is an alpha library for high performance machine learning research. Package 35.2k stars GitHub Back to Deep Learning