transformer_engine/jax/cpp_extensions.py · 1afb62569aa02dcbc4766cf2913be46f81363fbe · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

Make an import optional for a JAX API change that will happen soon. (#454) · d3157e2a
Frédéric Bastien authored Oct 03, 2023
```
Signed-off-by: Frederic Bastien <fbastien@nvidia.com>
```
d3157e2a

cpp_extensions.py 89.1 KB