# PyTorch

## Directory

### Classes

| Class | Description |
|-|-|
| [`flyteplugins.pytorch.Elastic`](https://www.union.ai/docs/v2/union/api-reference/integrations/pytorch/packages/flyteplugins.pytorch/elastic/page.md) | Elastic defines the configuration for running a PyTorch elastic job using torch. |

### Packages

| Package | Description |
|-|-|
| [`flyteplugins.pytorch`](https://www.union.ai/docs/v2/union/api-reference/integrations/pytorch/packages/flyteplugins.pytorch/_index) |  |

## Subpages

- [Classes](https://www.union.ai/docs/v2/union/api-reference/integrations/pytorch/classes/page.md)
- [Packages](https://www.union.ai/docs/v2/union/api-reference/integrations/pytorch/packages/page.md)

---
**Source**: https://github.com/unionai/unionai-docs/blob/main/content/api-reference/integrations/pytorch/_index.md
**HTML**: https://www.union.ai/docs/v2/union/api-reference/integrations/pytorch/