sagemaker.core.modules.train.container_drivers.distributed_drivers

sagemaker.core.modules.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.