Apex
Machine Learning Model TrainingML Frameworks

Apex

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

Open Source

About

This repository holds NVIDIA-maintained utilities to streamline mixed precision and distributed training in Pytorch. Some of the code here will be included in upstream Pytorch eventually. The intent of Apex is to make up-to-date utilities available to users as quickly as possible.

Open Source Health

Not enough history
Stars
8,999
Forks
1,526
License
BSD-3-Clause
Last commit
1 months ago
Python

Related Categories