Home / Python / jax jax a library for high-performance numerical computing with automatic differentiation and JIT compilation. Package 35.2k stars GitHub Back to Python
