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:
- Reverse-mode automatic differentiation in Rust (experiment)☆63Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- Documentation:☆120Updated 2 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 3 years ago
- Neat Bayesian machine learning examples☆58Updated 5 months ago
- Mathematical operations for JAX pytrees☆199Updated 6 months ago
- Understanding nuts and bolts of neural networks with PyTorch☆33Updated 4 years ago
- Auto Differentiate from scratch based on Autograd☆10Updated 3 years ago
- coding an autograd from scratch☆177Updated 6 years ago
- ☆85Updated 4 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- NumPy+Jax with named axes and an uncompromising attitude☆20Updated 3 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Extending JAX with custom C++ and CUDA code☆396Updated 10 months ago
- ☆33Updated 4 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 5 months ago
- Automatic Differentiation Handbook☆147Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Notebooks for the "Deep Learning with JAX" book☆149Updated 2 weeks ago
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated 11 months ago
- Progress, Notes, Summaries and a lot of Questions on Machine Learning☆55Updated 5 years ago
- Rudimentary automatic differentiation framework☆74Updated 6 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆481Updated this week
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆124Updated 2 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆267Updated last week
- JMP is a Mixed Precision library for JAX.☆203Updated 4 months ago
- Neural network from scratch in CUDA/C++☆80Updated 5 months ago
- Reinforcement learning library in JAX.☆100Updated last year