JAX

ML Framework

Visit site →

High-performance numerical computing and autodiff from Google. Used for research and with Flax for neural networks.

Reviews

No reviews yet.