JAX

August 23, 2025 2 weeks ago 1 min read

High performance numerical computing library with composable transformations like JIT compilation, vectorization, and automatic differentiation, widely used in research and large model training.