Torchtitan

A PyTorch native platform for training generative AI models