Alternatives and similar repositories for nanograd

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

• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆294Updated last year
• szmikler / pytorch-symbolic
  Symbolic API for model creation in PyTorch.
  ☆68Updated 7 months ago
• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆121Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated last year
• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆29Updated 4 years ago
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated this week
• jxbz / agd
  Automatic gradient descent
  ☆213Updated 2 years ago
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆104Updated 3 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆109Updated last year
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• IDSIA / modern-srwm
  Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep R…
  ☆172Updated 4 months ago
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆223Updated last year
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆156Updated 4 months ago
• kennysong / minigrad
  A minimal implementation of autograd (in pure Python) 🍰
  ☆95Updated 4 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆269Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆298Updated last month
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆180Updated 5 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆94Updated 10 months ago
• google-deepmind / nanodo
  ☆283Updated last year
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆422Updated 7 months ago
• bhosmer / mm
  ☆138Updated 11 months ago
• sradc / SmallPebble
  Small deep learning library written from scratch in Python, using NumPy/CuPy.
  ☆124Updated 3 years 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…
  ☆290Updated 10 months ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆47Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆40Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆52Updated last year