JAX

August 23, 2025 3 months 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.