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:
- coding an autograd from scratch☆179Updated 6 years ago
- Drop-in autodiff for NumPy.☆212Updated 3 weeks ago
- Yaae: Yet another autodiff engine (written in Numpy).☆27Updated 2 years ago
- Rudimentary automatic differentiation framework☆75Updated 6 years ago
- Documentation:☆121Updated 2 years ago
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆124Updated 2 years ago
- ☆85Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated last month
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆178Updated 3 months ago
- Notebooks for the "Deep Learning with JAX" book☆153Updated 2 months ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- ☆170Updated last year
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- JMP is a Mixed Precision library for JAX.☆208Updated 7 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆189Updated 8 months ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Mathematical operations for JAX pytrees☆200Updated 8 months ago
- Image augmentation library for Jax☆39Updated last year
- Understanding nuts and bolts of neural networks with PyTorch☆33Updated 4 years ago
- Symbolic API for model creation in PyTorch.☆67Updated 5 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆246Updated 5 years ago
- Extending JAX with custom C++ and CUDA code☆398Updated last year
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 4 years ago
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆36Updated 3 years ago
- Scalable Probabilistic Programming Library☆21Updated 7 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- ☆69Updated last year