evcu / numpy_autogradLinks
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
Sorting:
- coding an autograd from scratch☆177Updated 6 years ago
- Documentation:☆120Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Rudimentary automatic differentiation framework☆74Updated 6 years ago
- Auto Differentiate from scratch based on Autograd☆10Updated 2 years ago
- Mathematical operations for JAX pytrees☆198Updated 6 months ago
- ☆85Updated 4 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆106Updated 10 months ago
- ☆33Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆37Updated 3 years ago
- Neural Networks for JAX☆84Updated 8 months ago
- ☆67Updated last year
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- Image augmentation library for Jax☆39Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated last month
- Yaae: Yet another autodiff engine (written in Numpy).☆27Updated last year
- Learning some numerical linear algebra.☆70Updated 4 years ago
- ☆153Updated 5 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆175Updated last week
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 3 years ago
- Notebooks for the "Deep Learning with JAX" book☆148Updated 11 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆110Updated 3 years ago
- Progress, Notes, Summaries and a lot of Questions on Machine Learning☆55Updated 5 years ago
- High performance implementation of Deep neuroevolution in pytorch using mpi4py. Intended for use on HPC clusters☆27Updated 3 years ago