Skip to content

Enabling FP8 all-gather for TE Float8Tensor when using Torch FSDP2 #6404

Enabling FP8 all-gather for TE Float8Tensor when using Torch FSDP2

Enabling FP8 all-gather for TE Float8Tensor when using Torch FSDP2 #6404

Annotations

1 warning

JAX C++

succeeded Dec 5, 2024 in 21s