JAX

An open-source numerical computing library developed by Google that combines automatic differentiation with hardware acceleration, designed for high-performance machine learning research. JAX is particularly popular for large-scale model training and research into novel architectures, offering fine-grained control over computation that makes it well suited for pushing the boundaries of what is possible.
See also: pyTorch, tensorFlow, GPU.