Skip to content

[hardware, perf] feat: add TPUSplashAttention, SimpleFSDP, torch.compile, and DEFER_AND_FUSE to TorchTitanTPUEngine - #39

Merged
wuxibin89 merged 1 commit into
verl-project:mainfrom
jialei777:feat/tpu-torchtitan-compile-splash-simple-fsdp
Oct 8, 2026
Merged

[hardware, perf] feat: add TPUSplashAttention, SimpleFSDP, torch.compile, and DEFER_AND_FUSE to TorchTitanTPUEngine#39
wuxibin89 merged 1 commit into
verl-project:mainfrom
jialei777:feat/tpu-torchtitan-compile-splash-simple-fsdp

Commits

  1. Commits on Oct 7, 2026