Frameworks IntegrationΒΆ

In order to use SwitchML with one of the popular DNN frameworks such as Tensorflow or PyTorch, you need to use one of the integration methods.

Integration method

Supported Frameworks

Status

NCCL Plugin

Tensorflow through Horovod, PyTorch

Needs more testing

Pytorch Patch

Pytorch

Stable

Read more about each method by checking its corresponding documentation.