NVIDIA / TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

Date Created 2022-09-20 (2 years ago)

Commits 803 (last one 14 hours ago)

Stargazers 1,927 (4 this week)

Watchers 35 (0 this week)

Forks 321

License apache-2.0

Ranking

RepositoryStats indexes 579,555 repositories, of these NVIDIA/TransformerEngine is ranked #26,463 (95th percentile) for total stargazers, and #61,712 for total watchers. Github reports the primary language for this repository as Python, for repositories using this language it is ranked #4,027/115,123.

NVIDIA/TransformerEngine is also tagged with popular topics, for these it's ranked: python (#1,652/21870), deep-learning (#771/8344), machine-learning (#794/7877), pytorch (#477/5914), gpu (#84/895), cuda (#46/631)

Other Information

NVIDIA/TransformerEngine has 41 open pull requests on Github, 773 pull requests have been merged over the lifetime of the repository.

Github issues are enabled, there are 143 open issues and 204 closed issues.

There have been 13 releases, the latest one was published on 2024-10-08 (29 days ago) with the name v1.11.

Homepage URL: https://docs.nvidia.com/deeplearning/transformer-engine/user-guide/index.html

All Topics

fp8 gpu jax cuda python pytorch deep-learning machine-learning

Star History

Github stargazers over time

Watcher History

Github watchers over time, collection started in '23

Recent Commit History

803 commits on the default branch (main) since jan '22

Yearly Commits

Commits to the default branch (main) per year

Issue History

Languages

The primary language is Python but there's also others...

updated: 2024-11-06 @ 11:06pm, id: 539057023 / R_kgDOICFbfw