Infrastructure & Agents

What Is JAX?

JAX combines NumPy-like array programming with automatic differentiation and just-in-time compilation for fast computation. It is used for machine learning research and large-scale model training.