transformer_engine/jax/csrc/extensions.h · e30c36a30883c49b820096a0bb856c7ea71bebd5 · OpenDAS / TransformerEngine

jberchtold-nvidia authored Sep 30, 2025



Load modules during initialize
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
Co-authored-by: JAX Toolbox <jax@nvidia.com>

ce18bee7

extensions.h 6 KB