# Classes

| Class | Description |
|-|-|
| [`flytekitplugins.kfpytorch.task.CleanPodPolicy`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskcleanpodpolicy) |CleanPodPolicy describes how to deal with pods when the job is finished. |
| [`flytekitplugins.kfpytorch.task.Elastic`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskelastic) |Configuration for [`torch elastic training`](https://pytorch. |
| [`flytekitplugins.kfpytorch.task.ElasticWorkerResult`](../packages/flytekitplugins.kfpytorch.task#flytekitpluginskfpytorchtaskelasticworkerresult) |A named tuple representing the result of a torch elastic worker process. |
| [`flytekitplugins.kfpytorch.task.Master`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskmaster) |Configuration for master replica group. |
| [`flytekitplugins.kfpytorch.task.PyTorch`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskpytorch) |Configuration for an executable [`PyTorch Job`](https://github. |
| [`flytekitplugins.kfpytorch.task.PyTorchFunctionTask`](../packages/flytekitplugins.kfpytorch.task#flytekitpluginskfpytorchtaskpytorchfunctiontask) |Plugin that submits a PyTorchJob (see https://github. |
| [`flytekitplugins.kfpytorch.task.PytorchElasticFunctionTask`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskpytorchelasticfunctiontask) |Plugin for distributed training with torch elastic/torchrun (see. |
| [`flytekitplugins.kfpytorch.task.RestartPolicy`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskrestartpolicy) |RestartPolicy describes how the replicas should be restarted. |
| [`flytekitplugins.kfpytorch.task.RunPolicy`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskrunpolicy) |RunPolicy describes some policy to apply to the execution of a kubeflow job. |
| [`flytekitplugins.kfpytorch.task.Worker`](https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/packages/flytekitplugins.kfpytorch.task/page.md#flytekitpluginskfpytorchtaskworker) | |

---
**Source**: https://github.com/unionai/unionai-docs/blob/main/content/api-reference/plugins/kf-pytorch/classes.md
**HTML**: https://www.union.ai/docs/v1/flyte/api-reference/plugins/kf-pytorch/classes/