transformer_engine/jax/csrc/extensions/pybind.cpp · 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

pybind.cpp 9.01 KB