evcu / numpy_autograd
a simple implementation of autograd engine
☆24Updated 6 years ago
Alternatives and similar repositories for numpy_autograd:
Users that are interested in numpy_autograd are comparing it to the libraries listed below
- Documentation:☆119Updated last year
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- Rudimentary automatic differentiation framework☆74Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Auto Differentiate from scratch based on Autograd☆10Updated 2 years ago
- coding an autograd from scratch☆172Updated 6 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Mathematical operations for JAX pytrees☆198Updated 3 months ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- Image augmentation library for Jax☆38Updated 11 months ago
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆36Updated 3 years ago
- Notebooks for the "JAX in Action" book☆131Updated 8 months ago
- Better bindings for Python☆17Updated 2 years ago
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 3 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆88Updated 2 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Provides symbolic API for model creation in PyTorch.☆63Updated 6 months ago
- Stencil computations in JAX☆70Updated last year
- ☆21Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆468Updated last week
- Autograd extension for forward mode autodiff☆30Updated 7 years ago
- Building your own autograd mechanism based on PyTorch tensor only (not Variable, can be seen as numpy array)☆18Updated last year
- ☆165Updated 7 months ago
- Functional machine learning for fun☆84Updated 3 years ago
- Butterfly matrix multiplication in PyTorch☆167Updated last year