Efficiently computes derivatives of NumPy code.
7 daysStars7.5k24h gain-Score0.0
Find trending repositories by name or description.
Efficiently computes derivatives of NumPy code.
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
Implementation of a language-level autograd compiler
A deep learning framework built on an autograd engine with high level abstractions and low level control.