How do I implement and train a custom PyTorch model on Databricks using distributed training?

Suheb
Contributor

How can I build my own PyTorch machine-learning model and train it faster on Databricks by using multiple machines/GPUs instead of just one?