Add torch_compile flag for training networks#28

Open

wenxin0319 wants to merge 5 commits into

NVlabs:mainfrom

wenxin0319:main

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