tests/jax/test_sharding.py · 7976bd003fcf084dd068069b92a9a79b1743316a · OpenDAS / TransformerEngine

Tim Moon authored Jul 19, 2023



Replace deprecated sharding API
Signed-off-by: Tim Moon <tmoon@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

23f4864d

test_sharding.py 16.3 KB