JAX
Description
A numerical computing library developed by Google, JAX enables high-performance machine learning research with automatic differentiation and GPU/TPU support.
Features
Automatic differentiation for complex functions.
Just-In-Time (JIT) compilation for optimized performance.
Support for parallelization and vectorization.
Just-In-Time (JIT) compilation for optimized performance.
Support for parallelization and vectorization.
Website