meta-pytorch / torchftLinks
Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
☆474Updated 3 weeks ago
Alternatives and similar repositories for torchft
Users that are interested in torchft are comparing it to the libraries listed below
Sorting:
- PyTorch Single Controller☆953Updated this week
- Load compute kernels from the Hub☆389Updated this week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆596Updated 5 months ago
- Scalable and Performant Data Loading☆362Updated last week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆280Updated 2 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆732Updated this week
- A library to analyze PyTorch traces.☆464Updated last week
- Fast low-bit matmul kernels in Triton☆424Updated this week
- A Quirky Assortment of CuTe Kernels☆772Updated this week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆219Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- Applied AI experiments and examples for PyTorch☆314Updated 5 months ago
- ☆277Updated this week
- LLM KV cache compression made easy☆866Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆223Updated this week
- PyTorch-native post-training at scale☆605Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆255Updated this week
- ☆321Updated last year
- Pipeline Parallelism for PyTorch☆785Updated last year
- ☆178Updated last year
- Where GPUs get cooked 👩🍳🔥☆362Updated last week
- torchcomms: a modern PyTorch communications API☆323Updated this week
- ring-attention experiments☆165Updated last year
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆155Updated 2 years ago
- Cataloging released Triton kernels.☆291Updated 4 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆334Updated 3 months ago
- ☆344Updated 3 weeks ago
- kernels, of the mega variety☆657Updated 4 months ago
- Perplexity GPU Kernels☆554Updated 2 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆563Updated 2 weeks ago