torchx icon indicating copy to clipboard operation
torchx copied to clipboard

Add Azure Scheduler Support

Open d4l3k opened this issue 2 years ago • 0 comments

It would be nice to have Azure support so we can directly launch jobs on Azure's ML training service.

Example scheduler: AWS Batch https://github.com/pytorch/torchx/blob/main/torchx/schedulers/aws_batch_scheduler.py

Scheduler documentation: https://pytorch.org/torchx/main/schedulers

Azure Docs:

  • https://docs.microsoft.com/en-us/azure/machine-learning/how-to-set-up-training-targets
  • https://docs.microsoft.com/en-us/azure/machine-learning/how-to-train-distributed-gpu
  • https://docs.microsoft.com/en-us/azure/machine-learning/how-to-train-pytorch

d4l3k avatar Mar 07 '22 20:03 d4l3k