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:
- Documentation:☆121Updated 2 years ago
- coding an autograd from scratch☆179Updated 6 years ago
- Drop-in autodiff for NumPy.☆212Updated last week
- Notebooks for the "Deep Learning with JAX" book☆152Updated 2 months ago
- Reverse-mode automatic differentiation in Rust (experiment)☆63Updated 4 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Rudimentary automatic differentiation framework☆75Updated 6 years ago
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆125Updated 2 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- Mathematical operations for JAX pytrees☆200Updated 8 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated 3 months ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 weeks ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- ☆85Updated 4 years ago
- JMP is a Mixed Precision library for JAX.☆208Updated 6 months ago
- Evolution Strategy Library☆55Updated 5 years ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆41Updated 11 months ago
- Symbolic API for model creation in PyTorch.☆66Updated 4 months ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆119Updated last week
- Learning some numerical linear algebra.☆70Updated 4 years ago
- ☆68Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Yaae: Yet another autodiff engine (written in Numpy).☆27Updated 2 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆286Updated 2 weeks ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Image augmentation library for Jax☆39Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated last year
- ☆13Updated 4 years ago