Alternatives and similar repositories for grokfast:

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

• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆470Updated 11 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆178Updated 4 months ago
• microsoft / Samba
  Official implementation of "Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling"
  ☆833Updated last week
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆271Updated 2 months ago
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆614Updated 6 months ago
• Zyphra / BlackMamba
  Code repository for Black Mamba
  ☆234Updated 11 months ago
• KellerJordan / Muon
  Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead
  ☆219Updated 3 weeks ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆486Updated last week
• minyoungg / platonic-rep
  ☆495Updated 6 months ago
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆105Updated 3 months ago
• lucidrains / titans-pytorch
  Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch
  ☆891Updated this week
• nikhilvyas / SOAP
  ☆149Updated last month
• apple / ml-sigma-reparam
  ☆296Updated 7 months ago
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆146Updated 2 months ago
• SakanaAI / evo-memory
  Code to train and evaluate Neural Attention Memory Models to obtain universally-applicable memory systems for transformers.
  ☆280Updated 3 months ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆498Updated 3 months ago
• KellerJordan / cifar10-airbench
  94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds
  ☆196Updated 2 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆261Updated 7 months ago
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆171Updated last month
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆75Updated last month
• pbelcak / fastfeedforward
  A repository for log-time feedforward networks
  ☆217Updated 9 months ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆412Updated last month
• ekinakyurek / marc
  Public repository for "The Surprising Effectiveness of Test-Time Training for Abstract Reasoning"
  ☆285Updated 2 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆510Updated this week
• ClashLuke / HeavyBall
  Efficient optimizers
  ☆151Updated this week
• facebookresearch / coconut
  Training Large Language Model to Reason in a Continuous Latent Space
  ☆735Updated this week
• google-deepmind / nanodo
  ☆203Updated 6 months ago
• NX-AI / xlstm-jax
  Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…
  ☆75Updated 3 weeks ago
• Haiyang-W / TokenFormer
  [ICLR2025] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
  ☆482Updated this week
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆603Updated this week