Last released Aug 31, 2026
Profiles checkpointable PyTorch model blocks and selects a gradient-checkpointing plan for an activation-memory budget.