Repository navigation

Add ViT gradient checkpointing - #2058

Open

jaredrainsha wants to merge 6 commits into

lightly-ai:masterfrom

jaredrainsha:feat/2054-vit-grad-checkpointing

Open

Add ViT gradient checkpointing#2058
jaredrainsha wants to merge 6 commits into
lightly-ai:masterfrom
jaredrainsha:feat/2054-vit-grad-checkpointing

Commits on Sep 1, 2026
- feat(models): add ViT gradient checkpointing
  jaredrainsha
  committed
  87c5e46
Commits on Sep 15, 2026
- Merge branch 'master' into feat/2054-vit-grad-checkpointing
  gabrielfruet
  authored
  ceaecaf
Commits on Oct 23, 2026
- Merge branch 'master' into feat/2054-vit-grad-checkpointing
  gabrielfruet
  committed
  19712d2
- test: compare grads in ViT checkpointing test, import checkpoint_seq via module
  
  gabrielfruet
  and
  claude
  committed
  db1a4af
- test: assert checkpoint call count and intermediate count
  
  gabrielfruet
  and
  claude
  committed
  a90afb6
- fix: keep block grads under reentrant checkpointing with frozen inputs
  
  gabrielfruet
  and
  claude
  committed
  0c5094a