### Feature request An example of a model that uses multiple gpus at the same time, or we should have a way to make this easy to use ### Motivation _No response_ ### Other _No response_