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 6 years ago
- Drop-in autodiff for NumPy.☆214Updated this week
- Rudimentary automatic differentiation framework☆75Updated 6 years ago
- Notebooks for the "Deep Learning with JAX" book☆165Updated 6 months ago
- ☆85Updated 5 years ago
- Nvidia contributed CUDA tutorial for Numba☆264Updated 3 years ago
- Documentation:☆125Updated 2 years ago
- Yaae: Yet another autodiff engine (written in Numpy).☆28Updated 2 years ago
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆125Updated 3 years ago
- 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 7 months ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Reverse-mode automatic differentiation in Rust (experiment)☆63Updated 4 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Auto Differentiate from scratch based on Autograd☆10Updated 3 years ago
- Scalable Probabilistic Programming Library☆21Updated 7 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆125Updated last week
- A functional training loops library for JAX☆88Updated last year
- A minimal implementation of autograd (in pure Python) 🍰☆96Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 5 months ago
- Learning some numerical linear algebra.☆72Updated 4 years ago
- JMP is a Mixed Precision library for JAX.☆211Updated 10 months ago
- Turn SymPy expressions into trainable JAX expressions.☆356Updated 8 months ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- Neural Networks for JAX☆84Updated last year
- Linearly Constrained Separable Optimization☆49Updated 2 years ago
- ☆33Updated 4 years ago