Alternatives and similar repositories for nanograd

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

• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆120Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆46Updated last year
• szmikler / pytorch-symbolic
  Symbolic API for model creation in PyTorch.
  ☆68Updated 6 months ago
• jxbz / agd
  Automatic gradient descent
  ☆210Updated 2 years ago
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated this week
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆94Updated 9 months ago
• google-deepmind / nanodo
  ☆281Updated last year
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆269Updated last year
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆292Updated last year
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆37Updated 3 weeks ago
• google-deepmind / tf2jax
  ☆115Updated 2 weeks ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆167Updated 2 months ago
• BlackHC / neural_net_checklist
  ☆150Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• kennysong / minigrad
  A minimal implementation of autograd (in pure Python) 🍰
  ☆95Updated 4 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆29Updated 4 years ago
• bhosmer / mm
  ☆136Updated 10 months ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆54Updated 6 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆40Updated last year
• RobertRiachi / nanoPALM
  ☆144Updated 2 years ago
• sgrvinod / chess-transformers
  Teaching transformers to play chess
  ☆139Updated 8 months ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆166Updated this week
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆94Updated 4 months ago
• cloneofsimo / min-fsdp
  ☆89Updated last year
• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 3 years ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆123Updated last week