Repository navigation

Optional torch.compile and TF32 for training - #407

Merged

stephengreen merged 8 commits into

mainfrom

multi-gpu-optimizations

Sep 11, 2026

Merged

Optional torch.compile and TF32 for training#407
stephengreen merged 8 commits into
mainfrom
multi-gpu-optimizations

Commits on Sep 9, 2026
Commits on Sep 10, 2026
- Skip the no-graph-break compile test when the static-shape spline is not installed
  
  nihargupte-ph
  and
  claude
  committed
  ac889fc
Commits on Sep 11, 2026
- Address review: recompile on freeze_rb_layer change, drop_last on train loader only
  
  nihargupte-ph
  and
  claude
  committed
  ad501ac
- Re-lock uv.lock for torch>=2.6
  
  nihargupte-ph
  and
  claude
  committed
  cdc9772
- Docs: drop the glasflow fork note from the torch_compile section
  
  nihargupte-ph
  and
  claude
  committed
  5ad09da