meta-pytorch / torchftLinks
Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
☆467Updated 2 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☆359Updated this week
- PyTorch Single Controller☆939Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆278Updated last month
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆706Updated this week
- Scalable and Performant Data Loading☆360Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆592Updated 5 months ago
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- LLM KV cache compression made easy☆749Updated 3 weeks ago
- A library to analyze PyTorch traces.☆454Updated 3 weeks ago
- Applied AI experiments and examples for PyTorch☆312Updated 4 months ago
- Fast low-bit matmul kernels in Triton☆418Updated 3 weeks ago
- A Quirky Assortment of CuTe Kernels☆741Updated this week
- torchcomms: a modern PyTorch communications API☆319Updated this week
- ☆270Updated last week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆218Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆207Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆247Updated this week
- kernels, of the mega variety☆640Updated 3 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆328Updated 2 months ago
- Where GPUs get cooked 👩🍳🔥☆347Updated 3 months ago
- Cataloging released Triton kernels.☆282Updated 4 months ago
- ☆320Updated last year
- ☆224Updated last month
- ☆341Updated last week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆398Updated last week
- Pipeline Parallelism for PyTorch☆783Updated last year
- Perplexity GPU Kernels☆552Updated 2 months ago
- ☆178Updated last year
- ring-attention experiments☆161Updated last year
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆154Updated 2 years ago