https://github.com/NVIDIA/apex
- 总 Star 数量
- 9k
- 最近更新时间
- 2026-07-27
- 开源时间
- 2018-04-23
项目描述
Apex 是 NVIDIA 维护的 PyTorch 工具库，专注于混合精度训练和分布式训练优化。它提供融合版优化器、归一化层、自动混合精度等模块，利用 CUDA 融合内核加速计算，帮助深度学习开发者节省显存、提升训练速度。
2026.07.22–2026.07.28
2026.07.22–2026.07.28
Apex 是 NVIDIA 维护的 PyTorch 工具库，专注于混合精度训练和分布式训练优化。它提供融合版优化器、归一化层、自动混合精度等模块，利用 CUDA 融合内核加速计算，帮助深度学习开发者节省显存、提升训练速度。