srush / LLM-Training-Puzzles

Related projects: ⓘ

• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆966Updated this week
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆293Updated last year
• srush / Autodiff-Puzzles
  ☆325Updated 11 months ago
• EleutherAI / cookbook
  Deep learning for dummies. All the practical details and useful utilities that go into working with real models.
  ☆662Updated last month
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆345Updated 8 months ago
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆861Updated 5 months ago
• huggingface / nanotron
  Minimalistic large language model 3D-parallelism training
  ☆1,111Updated this week
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆452Updated last week
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆445Updated 6 months ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆807Updated this week
• PiotrNawrot / nanoT5
  Fast & Simple repository for pre-training and fine-tuning T5-style models
  ☆957Updated 3 weeks ago
• JonasGeiping / cramming
  Cramming the training of a (BERT-type) language model into limited compute.
  ☆1,284Updated 3 months ago
• cuda-mode / resource-stream
  CUDA related news and material links
  ☆1,079Updated 2 weeks ago
• databricks / megablocks
  ☆1,164Updated last week
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆492Updated this week
• mosaicml / examples
  Fast and flexible reference benchmarks
  ☆435Updated last month
• mosaicml / streaming
  A Data Streaming Library for Efficient Neural Network Training
  ☆1,076Updated this week
• mlfoundations / open_lm
  A repository for research on medium sized language models.
  ☆469Updated 3 weeks ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆321Updated 2 weeks ago
• TransformerLensOrg / TransformerLens
  A library for mechanistic interpretability of GPT-style language models
  ☆1,413Updated this week
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆1,489Updated this week
• llm-efficiency-challenge / neurips_llm_efficiency_challenge
  NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day
  ☆248Updated 10 months ago
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆341Updated last month
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,334Updated 2 months ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆451Updated last month
• HazyResearch / m2
  Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"
  ☆530Updated 3 months ago
• google-deepmind / recurrentgemma
  Open weights language model from Google DeepMind, based on Griffin.
  ☆592Updated 2 months ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆243Updated 10 months ago
• hao-ai-lab / LookaheadDecoding
  [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding
  ☆1,099Updated 7 months ago
• ayaka14732 / tpu-starter
  Everything you want to know about Google Cloud TPU
  ☆486Updated 2 months ago