NVIDIA/apex

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

8993 stars · 1525 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

NVIDIA-maintained PyTorch extension providing utilities for easy mixed precision and distributed training. It offers up-to-date CUDA and C++ extensions that eventually make their way into upstream PyTorch.

Use cases

• train deep learning models with mixed precision in pytorch
• speed up pytorch training on nvidia gpus
• run distributed training across multiple gpus
• use fused cuda kernels for faster optimizer steps
• reduce gpu memory usage during model training

When to choose

• you train large models on NVIDIA GPUs and need AMP/fused optimizer performance
• you need cutting-edge training utilities before they land in PyTorch
• you use NVIDIA NGC PyTorch containers and want matching extensions

When to avoid

• you only train on CPU or non-NVIDIA hardware
• you need a stable, simple setup without compiling CUDA extensions
• your use case is fully covered by native PyTorch AMP

Facets

library · maturity active

machine-learning llm-training gpu-computing deep-learning machine-learning gpu-computing python pytorch-extension mixed-precision distributed-training cuda-extensions nvidia linux gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="NVIDIA/apex")