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.