Commits · 26db7f34d04fcacbeba78e890c11e1403493c29a · OpenDAS / TransformerEngine

18 Apr, 2025 1 commit

[C][Jax] Move cuda kernels from Jax extensions to core (#1697) · 26db7f34

Kirthi Shankar Sivamani authored Apr 18, 2025



* Move jaxx cuda kernels to core
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



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

26db7f34