Alternatives and similar repositories for nanograd

Users that are interested in nanograd are comparing it to the libraries listed below

• sjmikler / pytorch-symbolic
  Symbolic API for model creation in PyTorch.
  ☆66Updated 4 months ago
• FlorianDietz / comgra
  A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…
  ☆286Updated 7 months ago
• BlackHC / neural_net_checklist
  ☆150Updated 11 months ago
• google-deepmind / nanodo
  ☆274Updated last year
• jxbz / agd
  Automatic gradient descent
  ☆208Updated 2 years ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆521Updated this week
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• sradc / SmallPebble
  Small deep learning library written from scratch in Python, using NumPy/CuPy.
  ☆124Updated 2 years ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆290Updated 10 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆385Updated 3 months ago
• kennysong / minigrad
  A minimal implementation of autograd (in pure Python) 🍰
  ☆96Updated 3 years ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆224Updated last year
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated 2 months ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆210Updated last year
• mle-infrastructure / mle-monitor
  Lightweight Experiment & Resource Monitoring 📺
  ☆187Updated 2 years ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆234Updated this week
• brohrer / sharpened-cosine-similarity
  An alternative to convolution in neural networks
  ☆256Updated last year
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆94Updated 7 months ago
• bpesquet / pyfit
  A minimalist neural networks library built on a tiny autograd engine
  ☆19Updated last year
• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆117Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• bhosmer / mm
  ☆131Updated 8 months ago
• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆29Updated 3 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆332Updated 2 years ago
• cair / tmu
  Implements the Tsetlin Machine, Coalesced Tsetlin Machine, Convolutional Tsetlin Machine, Regression Tsetlin Machine, and Weighted Tsetli…
  ☆147Updated 2 months ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆265Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆179Updated last month
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆101Updated this week
• modula-systems / modula
  🧱 Modula software package
  ☆204Updated 3 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆124Updated 11 months ago