Alternatives and similar repositories for picotron:

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

• huggingface / nanotron
  Minimalistic large language model 3D-parallelism training
  ☆1,701Updated this week
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆689Updated last week
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆524Updated last month
• srush / LLM-Training-Puzzles
  What would you do with 1000 H100s...
  ☆1,016Updated last year
• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆1,508Updated 4 months ago
• srush / awesome-o1
  A bibliography and survey of the papers surrounding o1
  ☆1,180Updated 4 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆232Updated 2 weeks ago
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆440Updated this week
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆368Updated 3 weeks ago
• KellerJordan / Muon
  Muon optimizer: +>30% sample efficiency with <3% wallclock overhead
  ☆505Updated last week
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆506Updated 4 months ago
• tspeterkim / flash-attention-minimal
  Flash Attention in ~100 lines of CUDA (forward pass only)
  ☆732Updated 2 months ago
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆464Updated last year
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆2,153Updated this week
• huggingface / lighteval
  Lighteval is your all-in-one toolkit for evaluating LLMs across multiple backends
  ☆1,313Updated this week
• mlfoundations / open_lm
  A repository for research on medium sized language models.
  ☆493Updated 2 months ago
• microsoft / Samba
  Official implementation of "Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling"
  ☆855Updated last month
• huggingface / search-and-learn
  Recipes to scale inference-time compute of open models
  ☆1,041Updated 3 weeks ago
• huggingface / picotron_tutorial
  ☆158Updated last month
• facebookresearch / coconut
  Training Large Language Model to Reason in a Continuous Latent Space
  ☆985Updated last month
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆600Updated 8 months ago
• haoliuhl / ringattention
  Large Context Attention
  ☆690Updated last month
• facebookresearch / blt
  Code for BLT research paper
  ☆1,436Updated last week
• zhuzilin / ring-flash-attention
  Ring attention implementation with flash attention
  ☆711Updated 3 weeks ago
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆307Updated 3 months ago
• efeslab / Nanoflow
  A throughput-oriented high-performance serving framework for LLMs
  ☆766Updated 6 months ago
• fla-org / flash-linear-attention
  🚀 Efficient implementations of state-of-the-art linear attention models in Torch and Triton
  ☆2,111Updated this week
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆365Updated last month