PABannier / nanograd

Related projects ⓘ

Alternatives and complementary repositories for nanograd

• sjmikler / pytorch-symbolic
  Provides symbolic API for model creation in PyTorch.
  ☆62Updated last month
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated last month
• sytelus / pcprep
  Scripts to prep PC for development use after OS installs
  ☆37Updated this week
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆174Updated 2 years ago
• jxbz / agd
  Automatic gradient descent
  ☆207Updated last year
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆50Updated last month
• vdesai2014 / inference-optimization-blog-post
  ☆82Updated 8 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆199Updated last year
• rasmusbergpalm / nanograd
  A miniscule implementation of reverse mode auto-differentiation
  ☆27Updated 3 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆95Updated 2 weeks ago
• nikhilvyas / SOAP
  ☆116Updated this week
• google-deepmind / nanodo
  ☆197Updated 3 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆151Updated 2 years ago
• google-deepmind / tf2jax
  ☆104Updated this week
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆84Updated 2 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 7 months ago
• cloneofsimo / min-fsdp
  ☆72Updated 4 months ago
• AakashKumarNain / diffusion_models
  All about the fundamentals and working of Diffusion Models
  ☆152Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆42Updated 5 months ago
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆118Updated 2 months ago
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆60Updated 10 months ago
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆112Updated 6 months ago
• hundredblocks / large-model-parallelism
  Functional local implementations of main model parallelism approaches
  ☆95Updated last year
• HomebrewNLP / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆46Updated 9 months ago
• KellerJordan / cifar10-airbench
  94% on CIFAR-10 in 2.67 seconds 💨 96% in 27 seconds
  ☆168Updated 3 weeks ago