jax

jax-ml/jaxFrameworks

A library for high-performance numerical computing with autodiff and JIT compilation.

More in Deep Learning

An end-to-end machine learning platform from Google.

201.2kupdated today2.21.0

Tensors and Dynamic neural networks in Python with strong GPU acceleration.

105kupdated today2.14.1

A high-level deep learning library with support for JAX, TensorFlow, and PyTorch backends.

64.4kupdated 1d ago3.12.4

Deep learning framework to train, deploy, and ship AI products Lightning fast.

31.4kupdated 12d ago2.6.6

PyTorch implementations of Stable Baselines (deep) reinforcement learning algorithms.

13.9kupdated 11d ago2.9.0

A standard API for reinforcement learning environments with popular reference environments (gym successor).

12.7kupdated today1.4.0