A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
☆165Aug 19, 2026Updated 2 months ago
Alternatives and similar repositories for torchsnapshot
Users that are interested in torchsnapshot are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆428Updated this week
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆247Aug 19, 2026Updated 2 months ago
- This repository contains the experimental PyTorch native float8 training UX☆226Aug 1, 2024Updated 2 years ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,265Sep 16, 2026Updated last month
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆179Dec 16, 2025Updated 10 months ago
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆997Sep 28, 2026Updated last month
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,078Apr 17, 2024Updated 2 years ago
- A library to analyze PyTorch traces.☆564Updated this week
- Pipeline Parallelism for PyTorch☆785Aug 21, 2024Updated 2 years ago
- A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.☆1,042Updated this week
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆362Oct 26, 2026Updated last week
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆551Updated this week
- A storage solution for PyTorch tensors with distributed tensor support.☆91Updated this week
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆252Jul 25, 2024Updated 2 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆66Mar 21, 2022Updated 4 years ago
- High performance model preprocessing library on PyTorch☆639Mar 29, 2024Updated 2 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆1,055Updated this week
- A tensor-aware point-to-point communication primitive for machine learning☆286Dec 17, 2025Updated 10 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,433Aug 21, 2025Updated last year
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated last year
- NCCL Profiling Kit☆156Jul 1, 2024Updated 2 years ago
- A resilient distributed training framework☆101Jul 24, 2026Updated 3 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Named tensors with first-class dimensions for PyTorch☆334Jun 14, 2023Updated 3 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆151Aug 23, 2025Updated last year
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆98Oct 21, 2026Updated last week
- ☆27Updated this week
- MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI tr…☆160Updated this week
- This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosys…☆46Oct 23, 2026Updated last week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆375Updated this week
- Tutel MoE: Optimized Mixture-of-Experts Library, Support GLM/Kimi/Gpt/DeepSeek/Qwen using FP8/NVFP4/MXFP4☆1,029Oct 21, 2026Updated last week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Test pytorch code with minimal computational overhead☆26Jun 8, 2023Updated 3 years ago
- PyTorch native quantization for training and inference☆3,018Updated this week
- Applied AI experiments and examples for PyTorch☆322Aug 22, 2025Updated last year
- Resource-adaptive cluster scheduler for deep learning training.☆459Mar 5, 2023Updated 3 years ago
- A validation and profiling tool for AI infrastructure☆400Updated this week
- A PyTorch native platform for training generative AI models☆5,871Updated this week
- PyTorch Single Controller☆1,084Updated this week