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☆179Updated 6 years ago
- Rudimentary automatic differentiation framework☆75Updated 6 years ago
- Drop-in autodiff for NumPy.☆213Updated last week
- Notebooks for the "Deep Learning with JAX" book☆162Updated 5 months ago
- Documentation:☆125Updated 2 years ago
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆124Updated 3 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 6 months ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- Yaae: Yet another autodiff engine (written in Numpy).☆28Updated 2 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Auto Differentiate from scratch based on Autograd☆10Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated last month
- Loopy belief propagation for factor graphs on discrete variables in JAX☆158Updated last year
- ☆71Updated 2 years ago
- JMP is a Mixed Precision library for JAX.☆211Updated 10 months ago
- ☆85Updated 4 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆312Updated this week
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Mathematical operations for JAX pytrees☆205Updated 11 months ago
- Source code, datasets and trained models for the paper Learning Advanced Mathematical Computations from Examples (ICLR 2021), by François…☆181Updated 4 years ago
- Annotating tensor shapes using Python types☆159Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆401Updated last week
- Extending JAX with custom C++ and CUDA code☆401Updated last year
- Image augmentation library for Jax☆40Updated last year
- Library for learning and inference with Sum-product Networks utilizing TensorFlow 2.x and Keras☆48Updated 2 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆125Updated 2 months ago