Add LGATr model#5

Open

spinjo wants to merge 9 commits intojet-universe:mainfrom

Commits on Feb 13, 2025

Add LGATr wrapper to networks/, and LGATr option in train_TopLandscape.sh
spinjo
committed

Commits on Feb 16, 2025

Commits on Feb 17, 2025

Support gradient checkpointing
spinjo
committed

Commits on Mar 15, 2025

Update based on changes in weaver-core
spinjo
committed

Commits on Mar 25, 2025

Turn amp off by default
spinjo
committed

Commits on Mar 30, 2025

Update GATr default configs
spinjo
committed

Commits on Jun 1, 2025

Changes based on recent lgatr rework
spinjo
committed