tensorplay.utils.checkpoint

torch.utils.checkpoint

`tensorplay.utils.checkpoint.checkpoint`	Runs `function(args, *kwargs)` eagerly (no recomputation).
`tensorplay.utils.checkpoint.checkpoint_sequential`	torch.utils.checkpoint.checkpoint_sequential compatibility.
`tensorplay.utils.checkpoint.CheckpointPolicy`
`tensorplay.utils.checkpoint.set_checkpoint_early_stop`	No-op for API parity.

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.stax

Static-graph optimization and acceleration. stax traces eager execution into a static graph, applies compiler passes (constant folding, dead code elimination, operator decomposition), and lowers the result to the availab

tensorplay.utils.data

tensorplay.utils.data.DataLoader

Ask DeepWiki