Alternatives and similar repositories for Triton-nanoGPT

Users that are interested in Triton-nanoGPT are comparing it to the libraries listed below

• cloneofsimo / min-fsdp
  ☆92Updated last year
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆186Updated last week
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 7 months ago
• berlino / seq_icl
  ☆53Updated last year
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated last year
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆251Updated 2 weeks ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆135Updated last year
• erfanzar / jax-flash-attn2
  A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…
  ☆34Updated 10 months ago
• VatsaDev / NanoPoor
  NanoGPT-speedrunning for the poor T4 enjoyers
  ☆73Updated 9 months ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆55Updated 4 months ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆196Updated 3 weeks ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆170Updated 6 months ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆80Updated 3 months ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆79Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆82Updated 2 years ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆131Updated last year
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆197Updated last year
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆53Updated last year
• Noumena-Network / nmoe
  MoE training for Me and You and maybe other people
  ☆331Updated 3 weeks ago
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆233Updated 2 weeks ago
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• NVIDIA-NeMo / Emerging-Optimizers
  ☆132Updated last week
• test-time-training / ttt-tk
  ☆44Updated 2 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆103Updated last month
• dvruette / barrel-rec-pytorch
  ☆53Updated 2 years ago
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆113Updated 4 months ago
• kyo-takano / chinchilla
  A toolkit for scaling law research ⚖
  ☆55Updated last year
• okarthikb / state-space-models
  ☆27Updated last year