Alternatives and similar repositories for scaling-book

Users that are interested in scaling-book are comparing it to the libraries listed below

• rwitten / HighPerfLLMs2024
  ☆547Updated last year
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆292Updated 4 months ago
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆584Updated last year
• microsoft / dion
  Dion optimizer algorithm
  ☆404Updated this week
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆214Updated 2 weeks ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆679Updated this week
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆585Updated 4 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆456Updated last week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA (+ more DSLs)
  ☆708Updated this week
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆552Updated last month
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆438Updated 9 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆153Updated 2 years ago
• meta-pytorch / monarch
  PyTorch Single Controller
  ☆921Updated this week
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆321Updated last month
• google-deepmind / nanodo
  ☆285Updated last year
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆687Updated this week
• marin-community / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆685Updated 3 weeks ago
• meta-pytorch / torchforge
  PyTorch-native post-training at scale
  ☆566Updated this week
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆658Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆436Updated this week
• srush / LLM-Training-Puzzles
  What would you do with 1000 H100s...
  ☆1,133Updated last year
• gpu-mode / reference-kernels
  Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!
  ☆174Updated 2 weeks ago
• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆2,170Updated last year
• huggingface / picotron
  Minimalistic 4D-parallelism distributed training framework for education purpose
  ☆1,917Updated 3 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆532Updated 4 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆177Updated last year
• srush / Autodiff-Puzzles
  ☆460Updated last year
• thinking-machines-lab / batch_invariant_ops
  ☆937Updated last month
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆148Updated last week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆298Updated last year