examples/jax/encoder/test_single_gpu_encoder.py · 4b2b39b4e1c64b841a8c361fccf046b15a4ad60f · 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

test_single_gpu_encoder.py 10.8 KB

Replace test_single_gpu_encoder.py