<#394 Feat: Add `ElasticConfig` message type for t...
# flyte-github
a
#394 Feat: Add `ElasticConfig` message type for torch elastic training Pull request opened by fg91 TL;DR This PR adds a custom proto for a torch elastic config for distributed training using torchrun. The elastic configuration set by the user in the task decorator is converted to the `ElasticPolicy` in kubeflow's `PytorchJob`. See flyteorg/flyte#3614 issue for more details. Type ☐ Bug Fix ☑︎ Feature ☐ Plugin Are all requirements met? ☑︎ Code completed ☑︎ Smoke tested ☐ Unit tests added ☐ Code documentation added ☑︎ Any pending items have an associated Issue Complete description Tracking Issue flyteorg/flyte#3614 Follow-up issue flyteorg/flyteidl ✅ All checks have passed 13/13 successful checks