Add torch_compile flag for training networks#28

Merged

juliusberner merged 5 commits into

NVlabs:mainfrom

wenxin0319:main

Jul 16, 2026

Commits on Jun 12, 2026

Add torch_compile_mode for training networks
wenxin0319
committed

Commits on Jun 13, 2026

Commits on Jun 17, 2026

fix calling compiler multiple times
wenxin0319
committed

Commits on Jul 15, 2026

Fix test fixture ordering to match trainer: on_train_begin before apply_torch_compile
wenxin0319
committed