rasmusbergpalm / nanogradLinks
A miniscule implementation of reverse mode auto-differentiation
β29Updated 4 years ago
Alternatives and similar repositories for nanograd
Users that are interested in nanograd are comparing it to the libraries listed below
Sorting:
- π¦ AD in less than 20 linesβ54Updated 4 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAXβ178Updated 3 months ago
- A JAX implementation of stochastic addition.β14Updated 3 years ago
- β17Updated last year
- AutoBound automatically computes upper and lower bounds on functions.β361Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β179Updated this week
- A functional training loops library for JAXβ88Updated last year
- Turn SymPy expressions into trainable JAX expressions.β348Updated 4 months ago
- Mathematical operations for JAX pytreesβ200Updated 9 months ago
- β60Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.β108Updated last year
- A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.β228Updated 10 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorchβ189Updated 9 months ago
- JAX Arrays for human consumptionβ106Updated 2 months ago
- Run PyTorch in JAX. π€β283Updated this week
- Documentation:β121Updated 2 years ago
- Files needed to replicate my research blogβ86Updated 10 months ago
- β105Updated last year
- Named Tensors for Legible Deep Learning in JAXβ201Updated last week
- OpTree: Optimized PyTree Utilitiesβ191Updated this week
- A Pytree Module system for Deep Learning in JAXβ214Updated 2 years ago
- β247Updated 2 months ago
- Parameter-Free Optimizers for Pytorchβ130Updated last year
- A simple library for scaling up JAX programsβ143Updated 10 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.β276Updated 2 weeks ago
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)β94Updated 9 months ago
- Dive into Jax, Flax, XLA and C++β31Updated 5 years ago
- JMP is a Mixed Precision library for JAX.β208Updated 7 months ago
- Riemannian Optimization Using JAXβ52Updated last year