Interface GradCheckpointRun

A gradient-checkpointed run of one training loop.

interface GradCheckpointRun {
    end(): void;
    releaseStep(): void;
    resume(): void;
    suspend(): void;
}

Index

Methods

end releaseStep resume suspend

Methods

end

end(): void
Switch checkpointing off and free everything it holds.

Returns void
- Defined in packages/inference/src/contracts/mlx/trainable.ts:69

releaseStep

releaseStep(): void
Free the checkpoints the last backward held (after its grads are eval'd).

Returns void
- Defined in packages/inference/src/contracts/mlx/trainable.ts:64

resume

resume(): void
Returns void
- Defined in packages/inference/src/contracts/mlx/trainable.ts:67

suspend

suspend(): void
Forward-only evaluation: switch checkpointing off / back on.

Returns void
- Defined in packages/inference/src/contracts/mlx/trainable.ts:66