Alternatives and similar repositories for nanograd:

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

• sytelus / pcprep
  Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.
  ☆40Updated last week
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆153Updated 2 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 9 months ago
• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆29Updated 3 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆41Updated 2 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 6 months ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆53Updated 3 weeks ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆232Updated last month
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆108Updated 2 years ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆63Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated 11 months ago
• google-deepmind / tf2jax
  ☆112Updated last month
• cgarciae / einop
  ☆60Updated 3 years ago
• jxbz / agd
  Automatic gradient descent
  ☆207Updated last year
• google-deepmind / nanodo
  ☆214Updated 8 months ago
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆58Updated last month
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 8 months ago
• sradc / SmallPebble
  Minimal deep learning library written from scratch in Python, using NumPy/CuPy.
  ☆121Updated 2 years ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 4 months ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated 3 weeks ago
• modula-systems / modula
  🧱 Modula software package
  ☆173Updated 2 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆120Updated 7 months ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆96Updated last month
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated 9 months ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆60Updated 3 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• apple / ml-ademamix
  ☆58Updated 4 months ago