transformer-engine
Install from the command line
Learn more about packages
$ docker pull ghcr.io/nvidia/transformer-engine:pytorch_te_hf_24.01-py3
Recent tagged image versions
- 59 Version downloads
Sorry, something went wrong.
Details
-
NVIDIA
- TransformerEngine
- Apache License 2.0
Last published
4 months ago
Discussions
14
Issues
128
Total downloads
59
Collaborators 5
-
ksivaman Kirthi Shankar Sivamani
-
timmoon10 Tim Moon
-
ptrendx Przemyslaw Tredak
-
cyanguwa Charlene Yang
-
mingxu1067 Ming-Xu Huang