Morph MORPH
SIGN IN SIGN UP
dhruvhead / ao UNCLAIMED
0 0 1 Python

COMMITS

/ torchao/testing/training/dtensor_utils.py
docs
July 9, 2025
V
mxfp8 training: add TP sharding strategy for dim1 kernel (#2436)
Vasiliy Kuznetsov committed
e675ffd
June 24, 2025
V
enable tensor parallelism for MXLinear (#2434)
Vasiliy Kuznetsov committed
32599be
V
enable to_mxfp8 cast for DTensor (#2420)
Vasiliy Kuznetsov committed
7d6bb6a
V
make dtensor shared test util more generic (#2416)
Vasiliy Kuznetsov committed
b963540
V
rename `torchao.testing.float8` to `torchao.testing.training` (#2415)
Vasiliy Kuznetsov committed
1a701e6