meta-pytorch / torchftLinks
Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
β454Updated 3 weeks ago
Alternatives and similar repositories for torchft
Users that are interested in torchft are comparing it to the libraries listed below
Sorting:
- Load compute kernels from the Hubβ337Updated last week
- π Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flashβ¦β271Updated last week
- PyTorch Single Controllerβ905Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.β584Updated 3 months ago
- A Quirky Assortment of CuTe Kernelsβ675Updated last week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.β640Updated this week
- Scalable and Performant Data Loadingβ345Updated this week
- A library to analyze PyTorch traces.β437Updated last week
- Fast low-bit matmul kernels in Tritonβ401Updated last week
- LLM KV cache compression made easyβ701Updated this week
- This repository contains the experimental PyTorch native float8 training UXβ226Updated last year
- Applied AI experiments and examples for PyTorchβ307Updated 3 months ago
- π Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.β217Updated last week
- β256Updated this week
- ring-attention experimentsβ160Updated last year
- kernels, of the mega varietyβ614Updated 2 months ago
- β316Updated last year
- torchcomms: a modern PyTorch communications APIβ295Updated this week
- Perplexity GPU Kernelsβ534Updated 3 weeks ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.β313Updated last month
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the β¦β239Updated this week
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA (+ more DSLs)β683Updated this week
- β177Updated last year
- PyTorch-native post-training at scaleβ549Updated this week
- Where GPUs get cooked π©βπ³π₯β319Updated 2 months ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and supβ¦β402Updated last week
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASSβ244Updated 6 months ago
- TPU inference for vLLM, with unified JAX and PyTorch support.β170Updated this week
- Cataloging released Triton kernels.β274Updated 2 months ago
- Pipeline Parallelism for PyTorchβ784Updated last year