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.
  ☆297Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆96Updated 6 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated last year
• jxbz / agd
  Automatic gradient descent
  ☆215Updated 2 years ago
• szmikler / pytorch-symbolic
  Symbolic API for model creation in PyTorch.
  ☆68Updated 8 months ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆47Updated last year
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆159Updated 3 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆307Updated last month
• bhosmer / mm
  ☆138Updated last year
• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆122Updated last year
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆53Updated last year
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆270Updated 2 years ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆176Updated this week
• google-deepmind / nanodo
  ☆285Updated last year
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆110Updated last year
• BlackHC / neural_net_checklist
  ☆210Updated last year
• 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…
  ☆293Updated 11 months ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆54Updated 9 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆125Updated 2 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated 2 weeks ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆39Updated last week
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆181Updated 6 months ago
• srush / Autodiff-Puzzles
  ☆460Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆188Updated last month
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆222Updated last year
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆97Updated 11 months ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆62Updated 2 years ago