Prevent CUDA OOM in PyTorch with dynamic lane switching
Summary
MEM v3 is a memory management tool for PyTorch that dynamically adjusts batch size and gradient accumulation to prevent CUDA OOM crashes during training and fine-tuning, featuring chaos resistance, crash-proof checkpoints, and live telemetry.
Similar Articles
Experimenting with an adaptive memory governor for PyTorch on an 8GB GPU — would love some feedback
The author is experimenting with an adaptive memory governor for PyTorch to prevent CUDA OOM errors on 8GB GPUs, sharing code and seeking community feedback.
@PyTorch: TRANSIT (TRANsparent Scale-In for multi-node Training) is a runtime that makes unified virtual memory practical for lar…
TRANSIT is a runtime that makes unified virtual memory practical for large-scale LLM training, reducing GPU usage by up to 50% without code changes, and will be presented at PyTorch Conference North America.
LRMoE: Demand-Side Load Re-Calibration for Efficient MoE Training on Memory-Constrained Nodes
LRMoE is a memory-aware MoE training system that jointly recalibrates backward demand and balances retained workload on memory-constrained GPU nodes, reducing cumulative MoE-layer training time by 21.2-44.8% on Qwen3-30B-A3B and GLM4.5-106B-A12B compared to SYMI.
@PyTorch: Use PyTorch-native libraries within the NVIDIA NeMo Framework to customize models to hit your exacting requirements for…
NVIDIA demonstrates how quantization-aware distillation (QAD) using NVIDIA Model Optimizer improves the Nemotron 3.5 Lightning model, reducing memory usage and increasing throughput while preserving accuracy for agentic benchmarks.
Keeping Large MoE Training Within Fixed GPU Memory (20 minute read)
This paper introduces methods to bound four key memory peaks in large Mixture-of-Experts training, enabling training at 1M context length with fixed GPU memory and up to 10.4× throughput improvement over baselines.