transformer_engine/jax/cpp_extensions.py · 2a86df2b2b1a09e635970f26658292c7019e9a84 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Canonicalize the dtype for the better user experience (#480) · 2a86df2b
zlsh80826 authored Oct 20, 2023
```
canonicalize the dtype for the better user experience
Signed-off-by: Reese Wang <rewang@nvidia.com>
```
2a86df2b

cpp_extensions.py 89.3 KB