Kubeflow MPI

Directory

Class	Description
`flytekitplugins.kfmpi.task.CleanPodPolicy`	CleanPodPolicy describes how to deal with pods when the job is finished.
`flytekitplugins.kfmpi.task.HorovodFunctionTask`	For more info, check out https://github.com/horovod/horovod.
`flytekitplugins.kfmpi.task.HorovodJob`	Configuration for an executable `Horovod Job using MPI operator`.
`flytekitplugins.kfmpi.task.Launcher`	Launcher replica configuration.
`flytekitplugins.kfmpi.task.MPIFunctionTask`	Plugin that submits a MPIJob (see https://github.com/kubeflow/mpi-operator) defined by the code within the _task_function to k8s cluster.
`flytekitplugins.kfmpi.task.MPIJob`	Configuration for an executable `MPI Job`.
`flytekitplugins.kfmpi.task.RestartPolicy`	RestartPolicy describes how the replicas should be restarted.
`flytekitplugins.kfmpi.task.RunPolicy`	RunPolicy describes some policy to apply to the execution of a kubeflow job.
`flytekitplugins.kfmpi.task.Worker`	Worker replica configuration.

Package	Description
`flytekitplugins.kfmpi.task`	This Plugin adds the capability of running distributed MPI training to Flyte using backend plugins, natively on Kubernetes.