Introduce configuration flags for gradient checkpointing#1564

Open

javak87 wants to merge 8 commits intoecmwf:developfrom

javak87:javad/dev/cond_checkpoint_all-1141-ver1

Commits on Jan 8, 2026

add cond_checkpoint helper func
Javad Kasravi
committed
add cond_checkpoint for pred head & embed
Javad Kasravi
committed
add gradient checkpoint flags
Javad Kasravi
committed
add cond_checkpoint to stream_embed_tras calss
Javad Kasravi
committed
add gradient checkpoint to engines
Javad Kasravi
committed
ruff the code
Javad Kasravi
committed
Merge branch 'develop' into javad/dev/cond_checkpoint_all-1141-ver1
javak87
authored
ruff the code
Javad Kasravi
committed