A PyTorch native platform for training generative AI models - View it on GitHub