Skip to content

Pull requests: NVIDIA/apex

New pull request New
74 Open 589 Closed
74 Open 589 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

fix groupnorm int32 index overflow
#1845 opened Sep 26, 2024 by tlogn Loading…
7
NCCLAllocator: Fix build failure
#1818 opened Jul 19, 2024 by Aidyn-A Loading…
3
Only print the warning message about TORCH_CUDA_ARCH_LIST if not set
#1812 opened Jun 27, 2024 by aurianer Loading…
fixup concats for grouped convolution
#1811 opened Jun 21, 2024 by techshoww Loading…
Avoid importing apex transformer automatically
#1801 opened May 8, 2024 by nWEIdia Loading…
4
fix building torch extension with glog
#1791 opened Mar 31, 2024 by petronny Loading…
1
Add xentropy bf16 support
#1790 opened Mar 29, 2024 by zyeric Loading…
64-bit indexing Adam
#1786 opened Mar 21, 2024 by cdm114514 Loading…
[CUDNN][cudnn-frontend] Bump cuDNN to 1.0.3
#1777 opened Feb 9, 2024 by eqy Loading…
1
Use master weights for bfloat16 FusedAdam when master_weights=True
#1731 opened Sep 22, 2023 by cbcase Loading…
3
Print the TORCH_CUDA_ARCH_LIST
#1705 opened Aug 7, 2023 by yncxcw Loading…
1
Make distributed fused lamb test names friendly to keyword filtering contrib
#1698 opened Jul 20, 2023 by crcrpar Loading…
Add custom build backend to support build args
#1692 opened Jul 6, 2023 by janEbert Loading…
6
A FasterRMSNorm implementation (based on FasterLayerNorm)
#1688 opened Jun 30, 2023 by Njuapp Loading…
1
Updating missing build dependency in pyproject.toml
#1680 opened Jun 13, 2023 by loadams Loading…
5
Calendar Versioning
#1648 opened Apr 21, 2023 by crcrpar • Draft
CUDA_HOME not checked before use
#1633 opened Apr 5, 2023 by x0152 Loading…
1
[sparse]update support for arbitrary N:M settings sparse
#1631 opened Apr 2, 2023 by LeiWang1999 Loading…
6
FusedAdam as a drop-in replacement for AdamW
#1615 opened Mar 14, 2023 by thomasw21 Loading…
fix bug when CUDA_HOME is None
#1610 opened Mar 13, 2023 by xiawu Loading…
3
Hack to load old distributed optimizer checkpoints
#1602 opened Mar 2, 2023 by timmoon10 • Draft 23.04
Remove deprecated features
#1589 opened Feb 15, 2023 by crcrpar • Draft
1
Fix incorrect initialization of GenericFusedScaleMaskSoftmax
#1586 opened Feb 15, 2023 by qmdnls Loading…
Instance norm nvfuser
#1582 opened Feb 9, 2023 by crcrpar • Draft
1
Fix FusedAdam.zero_grad(set_to_none=True)
#1579 opened Feb 6, 2023 by NouamaneTazi Loading…
4
ProTip! no:milestone will show everything without a milestone.