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
- 58 Version downloads
Sorry, something went wrong.
Details
- NVIDIA
- TransformerEngine
- Apache License 2.0
Last published
4 months ago
Discussions
14
Issues
127
Total downloads
58
Collaborators 5
- ksivaman Kirthi Shankar Sivamani
- timmoon10 Tim Moon
- ptrendx Przemyslaw Tredak
- cyanguwa Charlene Yang
- mingxu1067 Ming-Xu Huang