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
- 61 Version downloads
Loading
Sorry, something went wrong.
Details
-
NVIDIA
- TransformerEngine
- Apache License 2.0
- 1.8k stars
Last published
6 months ago
Discussions
14
Issues
159
Total downloads