cosmos

Hi, thanks for your great work. I was just wondering if it's possible to add support for training the Cosmos model with 2 billion parameters for images. There's also a model for video. It's a very, very cool and powerful model. Regards.

The issue names no files, tests, or entry points. First map the existing model-training entry points and determine the scope of Cosmos support, including the 2-billion-parameter image model and the separately mentioned video model; done should mean the agreed training targets are supported and covered by tests.

Written by the indexing model from the issue text.