tests/jax/distributed_test_base.py · d2d4cf9142d522562f6e05b2e5768619bd6a1356 · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Jun 13, 2024



* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9416519d

distributed_test_base.py 4.21 KB

Replace distributed_test_base.py