COMMITS
/ torchao/float8/float8_scaling_utils.py February 22, 2025
V
delete delayed scaling from torchao.float8 (#1753)
Vasiliy Kuznetsov committed
February 10, 2025
D
Support power of 2 scaling factors in float8 training and use e4m3 everywhere (#1670)
Daniel Vega-Myhre committed
January 13, 2025
V
[cleanup][1/x] make hp_tensor_to_float8_dynamic only work with hp inputs (#1458)
Vasiliy Kuznetsov committed
December 4, 2024
L
[float8] Allow specifying arbitrary dtype for each tensor (#1378)
Luca Wehrstedt committed
November 22, 2024
V
float8 training: fix bug with AC + compile (#1329)
Vasiliy Kuznetsov committed
V
fix bug with float8 training + FSDP2 + TP (#1327)
Vasiliy Kuznetsov committed
November 11, 2024
A
Ruff Lint fixes (torchao/float8) (#1239)
Apurva Jain committed
October 7, 2024
V
float8 training axiswise scaling support with per-gemm-argument configuration (#940)
Vasiliy Kuznetsov committed
V
add axiswise granularity to Float8Tensor (#919)
Vasiliy Kuznetsov committed
September 26, 2024
W
[float8] all-reduce amax on dp mesh instead of global pg (#933)
Wei (Will) Feng committed
August 28, 2024
V
float8 training: add static scaling (#760)
Vasiliy Kuznetsov committed
July 30, 2024
V
move float8_experimental to torchao/float8
Vasiliy Kuznetsov committed