evcu / numpy_autogradLinks
a simple implementation of autograd engine
☆24Updated 7 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☆180Updated 7 years ago
- Drop-in autodiff for NumPy.☆213Updated last month
- A minimalist deep learning library written from scratch in Python☆132Updated 2 weeks ago
- Reverse-mode automatic differentiation in Rust (experiment)☆63Updated 5 years ago
- Rudimentary automatic differentiation framework☆75Updated 6 years ago
- Notebooks for the "Deep Learning with JAX" book☆168Updated 8 months ago
- Documentation:☆126Updated 2 years ago
- ☆85Updated 5 years ago
- ☆68Updated 10 months ago
- Module 0 - Fundamentals☆110Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 9 months ago
- Source code, datasets and trained models for the paper Learning Advanced Mathematical Computations from Examples (ICLR 2021), by François…☆181Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 6 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆190Updated 3 weeks ago
- Example python package with pybind11 cpp extension☆57Updated 5 years ago
- A minimal implementation of autograd (in pure Python) 🍰☆96Updated 4 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆130Updated last week
- ☆71Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- Training MLP on MNIST in 1.5 seconds with pure CUDA☆46Updated last year
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago
- Image augmentation library for Jax☆41Updated last year
- Scalable Probabilistic Programming Library☆21Updated 7 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Repository of models in Pyro☆30Updated last year
- Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach☆137Updated last year
- Mathematical operations for JAX pytrees☆207Updated last year
- Learning some numerical linear algebra.☆72Updated 5 years ago