torch_remat fine-grained activation checkpointing API
☆15Jul 20, 2026Updated this week
Alternatives and similar repositories for remat
Users that are interested in remat are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosys…☆34Updated this week
- Memory Efficient Training Framework for Large Video Generation Model☆25Apr 22, 2024Updated 2 years ago
- ☆21Mar 3, 2025Updated last year
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆89Updated this week
- ☆13Jul 12, 2026Updated last week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Seemless interface of using PyTOrch distributed with Jupyter notebooks☆59Sep 15, 2025Updated 10 months ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated last year
- Official repository for the paper DynaPipe: Optimizing Multi-task Training through Dynamic Pipelines☆19Dec 8, 2023Updated 2 years ago
- Framework for Algorithmic Correctness Testing of Operators☆16Mar 9, 2026Updated 4 months ago
- DL Dataloader Benchmarks☆20Jan 27, 2025Updated last year
- ☆16Sep 24, 2024Updated last year
- GPU accelerated Perlin Noise in python☆11Oct 23, 2020Updated 5 years ago
- Slowdown prediction module of Echo: Simulating Distributed Training at Scale☆13Jul 11, 2026Updated last week
- Code repository for the SOSP'25 paper DCP: Addressing Input Dynamism In Long-Context Training via Dynamic Context Parallelism.☆21Nov 28, 2025Updated 7 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Cellular Automata - Pokemon Type Battle Simulation☆11Oct 26, 2024Updated last year
- ☆13Aug 12, 2024Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- ☆33Mar 1, 2023Updated 3 years ago
- ☆17Jan 30, 2023Updated 3 years ago
- utilities to facilitate working with codebases that don't ascribe to normal package management paradigms, e.g. ML research code that can …☆13Nov 26, 2022Updated 3 years ago
- Convolutions and more as einsum for PyTorch☆18Jun 6, 2024Updated 2 years ago
- PyTorch centric eager mode debugger☆46Dec 16, 2024Updated last year
- Counterfactual Evaluation and Learning for Interactive Systems: Foundations, Implementations, and Recent Advances☆12Aug 14, 2022Updated 3 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Using DTensor on Google Cloud☆18Sep 18, 2022Updated 3 years ago
- PyTorch Lightning based framework to run experiments for self-supervised learning tasks.☆10Feb 14, 2020Updated 6 years ago
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆58May 3, 2026Updated 2 months ago
- Tabular data imputation and generation, with flexible modeling of quantitative features via hierarchical binning (TMLR, 2025)☆16Mar 10, 2025Updated last year
- ☆56Apr 30, 2025Updated last year
- ☆15May 20, 2026Updated 2 months ago
- Reinforcement Learning via Regressing Relative Rewards☆40Dec 12, 2024Updated last year
- Simple PyTorch graph capturing.☆21May 31, 2023Updated 3 years ago
- Templates for commonly used GitHub actions steps☆13Dec 13, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Simple repository contribution statistics☆15Updated this week
- Mirroring displays on Linux☆13Aug 22, 2016Updated 9 years ago
- DPO, but faster 🚀☆52Dec 6, 2024Updated last year
- Matlab toolbox to compute the statistics, pdf, cdf, inverse cdf and random numbers of the generalized chi-square distribution.☆12Jul 5, 2026Updated 2 weeks ago
- ☆12Mar 7, 2022Updated 4 years ago
- Official FlashDecoder Github☆15Apr 4, 2026Updated 3 months ago
- a whirlwind tour to deep learning and deep learning systems☆81Updated this week