torch_remat fine-grained activation checkpointing API
☆15Jul 28, 2026Updated last 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…☆35Updated 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.☆93Updated this week
- ☆13Jul 12, 2026Updated 3 weeks ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated 2 years ago
- Communication patterns for AI, built on top of NCCL device and host APIs☆25Updated this week
- ☆10Sep 4, 2021Updated 4 years ago
- 清华大学宿舍洗衣机空闲提醒小程序☆14Feb 4, 2021Updated 5 years ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- 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 5 months ago
- DL Dataloader Benchmarks☆20Jan 27, 2025Updated last year
- 在线图书借阅系统 - 2017 THU OOP课大作业☆14Jul 1, 2018Updated 8 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- GPU accelerated Perlin Noise in python☆11Oct 23, 2020Updated 5 years ago
- Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…☆29Apr 17, 2024Updated 2 years ago
- A dispatcher based on Hashicorp's Raft for Casbin.☆17May 15, 2026Updated 2 months ago
- Unofficial implementation of paper : Exploring the Space of Key-Value-Query Models with Intention☆12May 24, 2023Updated 3 years ago
- ☆14Aug 12, 2024Updated last year
- Cellular Automata - Pokemon Type Battle Simulation☆11Oct 26, 2024Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- CLI to manage compute via our API☆16Jul 7, 2026Updated last month
- A collection of various custom nodes for ComfyUI (Work in progress)☆15Jun 9, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- FGSM and L-BFG implementation☆16Jul 15, 2021Updated 5 years ago
- 清华成绩刮刮乐☆20Feb 17, 2014Updated 12 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
- Simulating Distributed Training at Scale☆15Sep 15, 2025Updated 10 months ago
- access ChatGPT/Gemini/Claude from Emacs without APIs☆12Jun 11, 2026Updated last month
- 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
- PyTorch Lightning based framework to run experiments for self-supervised learning tasks.☆10Feb 14, 2020Updated 6 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆58May 3, 2026Updated 3 months ago
- Tabular data imputation and generation, with flexible modeling of quantitative features via hierarchical binning (TMLR, 2025)☆16Mar 10, 2025Updated last year
- Experimental CUDA kernel framework unifying typed dimensions, NVRTC JIT specialization, and ML‑guided tuning.☆47Feb 9, 2026Updated 6 months ago
- Sequence-level 1F1B schedule for LLMs.☆37Aug 26, 2025Updated 11 months ago
- ☆16Jul 30, 2026Updated last week
- ☆14Sep 4, 2020Updated 5 years ago
- Reinforcement Learning via Regressing Relative Rewards☆40Dec 12, 2024Updated last year