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
- Documentation:β122Updated 2 years ago
- A JAX implementation of stochastic addition.β14Updated 3 years ago
- β17Updated last year
- A functional training loops library for JAXβ88Updated last year
- β60Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β181Updated this week
- Mathematical operations for JAX pytreesβ200Updated 9 months ago
- β247Updated 2 months ago
- Dive into Jax, Flax, XLA and C++β32Updated 5 years ago
- Parameter-Free Optimizers for Pytorchβ130Updated last year
- Turn SymPy expressions into trainable JAX expressions.β349Updated 5 months ago
- JAX Arrays for human consumptionβ106Updated 2 months ago
- Riemannian Optimization Using JAXβ52Updated last year
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processesβ¦β246Updated 5 years ago
- A Python package of computer vision models for the Equinox ecosystem.β109Updated last year
- β105Updated last year
- Pytorch implementation of SuperPolyak subgradient method.β43Updated 2 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAXβ180Updated 4 months ago
- A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.β229Updated 11 months ago
- An introduction to functional programming for scalable statistical computingβ75Updated last year
- Open source code for EigenGame.β30Updated 2 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorchβ189Updated 9 months ago
- Multiple dispatch over abstract array types in JAX.β129Updated last week
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasβ¦β24Updated 2 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.β276Updated last week
- Named Tensors for Legible Deep Learning in JAXβ203Updated this week
- Inference Combinators in JAXβ50Updated 4 months ago
- Neural Networks for JAXβ84Updated last year
- β115Updated 2 weeks ago