JAX

An open-source Python library developed by Google for high-performance numerical computing and machine learning, providing automatic differentiation, just-in-time (JIT) compilation, vectorization, and execution on CPUs, GPUs, and TPUs.

Explore more about Libraries

NumPy

A Python library for efficient numerical computing with multidimensional arrays, mathematical operations, and scientific data processing.

TensorFlow

An open-source machine learning framework for building, training, and deploying deep learning models.

PyTorch

An open-source machine learning framework designed for deep learning, tensor computation, and dynamic neural network construction.

Related terms

Signal, not noise.