transformer_engine/jax/cpp_extensions.py · d3157e2a7e45839895e11aab836ff2990b78f162 · OpenDAS / TransformerEngine · GitLab

"csrc/vscode:/vscode.git/clone" did not exist on "1f5ab1c5f0a99fd2e03ad88fccc6c9112b781791"

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