Unrolling thread...
bit confused bout CP and TP. we cant scale with FSDP on