NVIDIA/apex

View on GitHub

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Stars: 9k
Forks: 1.5k
Open beginner issues: 0
Indexed issues: 680
Avg merge: 2d 4h
Merged PRs (30d): 3

Dominant language: Python
License: BSD-3-Clause
Last GitHub push: Sep 14, 2026
Latest indexed: Sep 19, 2026
Contributing guide: No contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

Beginner friendly Everything

Open All

680 open issues indexed

`torch.cuda.amp` still used in 5 files (incl. private `autocast_mode._cast` / `grad_scaler.OptState`) — deprecated since torch 2.3/2.4, no guards Open

Difficulty 4/5 3-5 days Newbie friendliness 64/100

NVIDIA/apex#2024 · Opened Aug 20, 2026
incompatibility with nightly pytorch as it requires C++20 Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

NVIDIA/apex#2005 · Opened May 22, 2026
[Bug] FusedRMSNorm leaks 2 CUDA tensors per forward call under torch.no_grad() Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 48/100

NVIDIA/apex#1999 · Opened Apr 29, 2026
CWE-22/CWE-73 in permutation cache path: APEX_ASP_CACHE_DIR controls write destination Open

Difficulty 3/5 1-2 days Newbie friendliness 55/100

NVIDIA/apex#1998 · Opened Apr 27, 2026
cannot import name 'amp' from 'apex' Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1991 · 1 comment · Opened Apr 2, 2026
problem size question for multi-tensor-axpby Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

NVIDIA/apex#1970 · Opened Dec 15, 2025
Reduce the number of ignored rules Open

NVIDIA/apex#1964 · 1 reaction · 2 assignees · Opened Nov 29, 2025
Migrate to LibTorch Stable ABI Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1946 · 1 comment · Opened Nov 7, 2025
Think about removing `apex_C` Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

NVIDIA/apex#1945 · 1 comment · Opened Nov 7, 2025
Unable to install apex-ModuleNotFoundError: No module named 'packaging' Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1939 · 2 comments · Opened Oct 29, 2025
适用于：“VS2022 ” “CUDA126” ”python 3.12“ “ComfyUI:-v0.3.40" APEX编辑的"setup.py"代码“ Open

windows

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1909 · 1 comment · 1 reaction · Opened Jun 9, 2025
Suitable for: "VS2022" "CUDA126" "Python 3.12" "ComfyUI: - v0.3.40" APEX edited "setup. py" code“ Open

windows

Difficulty 4/5 3-5 days Newbie friendliness 32/100

NVIDIA/apex#1908 · Opened Jun 9, 2025
Suitable for: "VS2022" "CUDA126" "Python 3.12" "ComfyUI: - v0.3.40" APEX edited "setup. py" code“ Open

windows

Difficulty 4/5 3-5 days Newbie friendliness 22/100

NVIDIA/apex#1907 · 1 comment · Opened Jun 9, 2025
Support env vars to specify cuda extensions to build Open

Difficulty 4/5 3-5 days Newbie friendliness 30/100

NVIDIA/apex#1906 · Opened May 31, 2025
How to solve the incompatibility problem between apex and zero shot tts Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 20/100

NVIDIA/apex#1899 · 1 comment · Opened Apr 28, 2025
apex @ nemo v2.2.1 version: build fails: ATen/Dispatch.h:199:48: error: cannot convert ‘const at::DeprecatedTypeProperties’ to ‘c10::ScalarType’ Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1897 · 4 comments · Opened Apr 23, 2025
Compilation Error in Apex with CUDA 12.6: isfinite< ::c10::Half> Undefined in Device Code Open

windows

Difficulty 4/5 3-5 days Newbie friendliness 32/100

NVIDIA/apex#1885 · Opened Feb 26, 2025
AttributeError: module 'torch' has no attribute 'library' Open

Difficulty 4/5 3-5 days Newbie friendliness 20/100

NVIDIA/apex#1881 · 2 comments · Opened Feb 9, 2025
install warning Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 20/100

NVIDIA/apex#1878 · 1 comment · Opened Jan 25, 2025
[Orin Nano] Error when trying to use sparsity Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1874 · 1 comment · Opened Jan 8, 2025
as: symbol lookup error: as: undefined symbol: ZSTD_compressStream2 Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1871 · 1 comment · Opened Jan 4, 2025
How should I search for the corresponding APEX release for a specific PyTorch version? Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1868 · 3 comments · Opened Dec 30, 2024
AttributeError: module 'torch.compiler' has no attribute 'is_compiling' Open

bug

Difficulty 2/5 1-3 hours Newbie friendliness 35/100

NVIDIA/apex#1857 · 6 comments · 1 reaction · Opened Nov 15, 2024
Failed the last time, succeeded the next time?上一次还失败，下一次就成功了？ Open

Difficulty 4/5 3-5 days Newbie friendliness 20/100

NVIDIA/apex#1856 · 5 comments · Opened Nov 1, 2024
FusedRMSNormAffineMixedDtypesFunction is not importable in the PyTorch build without distributed support Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 45/100

NVIDIA/apex#1853 · Opened Oct 28, 2024
关于解决ModuleNotFoundError: No module named 'torch'导致安装失败 Open

bug windows

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1852 · 30 comments · Opened Oct 24, 2024
How to install apex Open

windows

Difficulty 3/5 1-2 days Newbie friendliness 20/100

NVIDIA/apex#1851 · 17 comments · 3 reactions · Opened Oct 19, 2024
到底要怎样才能安装apex Open

Difficulty 2/5 1-3 hours Newbie friendliness 15/100

NVIDIA/apex#1850 · 1 comment · Opened Oct 11, 2024
AdamW implementation does not truly decouple learning rate and weight decay Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 42/100

NVIDIA/apex#1849 · 2 comments · Opened Oct 9, 2024
Failed to build installable wheels for some pyproject.toml based projects (apex) Open

windows

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1848 · 5 comments · Opened Oct 3, 2024
Literature associated with fused_dense Open

Difficulty 2/5 1-3 hours Newbie friendliness 35/100

NVIDIA/apex#1846 · 1 comment · Opened Sep 27, 2024
ASP Automatic Sparsity forward function For Loop Error Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1843 · 1 comment · Opened Sep 6, 2024
Discrepancy with Optimizer States and Model State Dict when using store_param_remainders==True Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1842 · 8 comments · Opened Sep 5, 2024
Gradient Overflow with Specific GPU Combinations in Multi-GPU Setup (NVIDIA RTX 3090) Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

NVIDIA/apex#1841 · Opened Sep 5, 2024
No module named 'amp_C' Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 25/100

NVIDIA/apex#1840 · Opened Sep 4, 2024
loss scale Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

NVIDIA/apex#1839 · Opened Sep 4, 2024
How to improve training performance with Apex package Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

NVIDIA/apex#1838 · Opened Aug 31, 2024
install bug with pytorch2.0.1 Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1834 · 1 comment · Opened Aug 27, 2024
Installation with Cuda extentions is failling Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 28/100

NVIDIA/apex#1831 · 5 comments · Opened Aug 20, 2024
No CUDA runtime is found, using CUDA_HOME='/home/shengjieyi/cuda1108' . Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 25/100

NVIDIA/apex#1828 · Opened Aug 16, 2024
Unsuccessful installation of apex library. (Preparing metadata (pyproject.toml) did not run successfully.) Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 25/100

NVIDIA/apex#1827 · 4 comments · 1 reaction · Opened Aug 15, 2024
Slow Performance with "Exhaustive Search" Permutation Strategy for Channel Pruning in CNN Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 28/100

NVIDIA/apex#1826 · 4 reactions · Opened Aug 14, 2024
Unable to install Apex Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 30/100

NVIDIA/apex#1824 · 3 comments · Opened Aug 6, 2024
Install set.up Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1822 · Opened Jul 25, 2024
Unknown CUDA arch (compute) or GPU not supported error while installing on docker ubuntu with cuda 12.1 Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1819 · 1 comment · Opened Jul 19, 2024
Unable to install Apex on Linux(debian) with CUDA 12.1 and torch 2.2.2 Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1817 · 2 comments · 1 reaction · Opened Jul 14, 2024
unable to install Open

Difficulty 2/5 1-3 hours Newbie friendliness 25/100

NVIDIA/apex#1815 · 5 comments · 2 reactions · Opened Jul 1, 2024
Unable to install Apex Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 35/100

NVIDIA/apex#1810 · 2 comments · 2 reactions · Opened Jun 21, 2024
Win11+Visual Studio 2022,install successfully. Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

NVIDIA/apex#1809 · 3 comments · Opened Jun 18, 2024
Error Open

Difficulty 3/5 1-2 days Newbie friendliness 10/100

NVIDIA/apex#1807 · Opened Jun 16, 2024

NVIDIA/apex

Get new issues in your inbox