AutodiffCheckpointType

Enumeration for JAX checkpoint types.

JAX (Core)PaxML / Praxis

JAX (Core)

API: jax.remat

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: praxis.layers.AutodiffCheckpointType

Strategy: Direct Mapping

Official Docs ↗