JAX
A Python library from Google for high-performance numerical computing with autodiff, JIT compilation, and vectorization via a NumPy-like API.
Deep Learning
A Python library from Google for high-performance numerical computing with autodiff, JIT compilation, and vectorization via a NumPy-like API.