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
- Auto Differentiate from scratch based on Autograd☆10Updated 2 years ago
- coding an autograd from scratch☆172Updated 6 years ago
- Mathematical operations for JAX pytrees☆198Updated 3 months ago
- Reverse-mode automatic differentiation in Rust (experiment)☆62Updated 4 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 3 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
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- ☆85Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems☆19Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 8 months ago
- Neural Networks for JAX☆83Updated 6 months ago
- 🦠 AD in less than 20 lines☆54Updated 3 years ago
- Notebooks for the "JAX in Action" book☆134Updated 9 months ago
- Functional machine learning for fun☆84Updated 3 years ago
- Symbolic API for model creation in PyTorch.☆65Updated last week
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- Image augmentation library for Jax☆39Updated 11 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- ☆95Updated 3 weeks ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last month
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- Rudimentary automatic differentiation framework☆74Updated 5 years ago
- Module 0 - Fundamentals☆101Updated 6 months ago