transformer_engine/jax/fp8.py · 0d25199158abe9167ecf5ceec655dab24c7c626e · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

WAR a JAX issue (#221) · 0d251991
Frédéric Bastien authored May 16, 2023
```
Signed-off-by: Frederic Bastien <fbastien@nvidia.com>
```
0d251991

fp8.py 15.4 KB