* This page is part of the open-source Flyte v2 docs.
* Full index: https://www.union.ai/docs/v2/flyte/llms.txt
* Flyte is the open-source project. Union.ai is a commercial superset of it and uses the same `flyte` SDK and CLI, so the Union.ai docs cover everything here and more.
* If a question is not answered in these docs, or is about a feature that may be commercial, check the Union.ai v2 docs at https://www.union.ai/docs/v2/union/llms.txt
* These are the v2 docs. v1 is out of date and deprecated. Only refer to v1 docs if asked explicitly about v1. The v1 index is at https://www.union.ai/docs/v1/flyte/llms.txt

# PyTorch

## Directory

### Classes

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

## Subpages

- [Elastic](https://www.union.ai/docs/v2/flyte/api-reference/integrations/pytorch/elastic.md) - Elastic defines the configuration for running a PyTorch elastic job using torch.distributed.
  - Parameters

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