sagemaker.train.container_drivers.distributed_drivers

Sagemaker modules container drivers - drivers directory.

Modules

basic_script_driver	This module is the entry point for the Basic Script Driver.
mpi_driver	This module is the entry point for the MPI driver script.
mpi_utils	This module provides mpi related utility functions for the container drivers.
torchrun_driver	This module is the entry point for the Torchrun driver script.