jax-ml/jax
JAX
JAX is a library for accelerated computing, autodiff, and transformations of NumPy-like programs.
Catalog projects marked with #autodiff. Tags work as dedicated landing pages, so related tools are easier to find and connect.
This collection holds 1 projects with a combined 35,919 GitHub stars. Main languages: Python.