Our inference and training framework to run on the Cosmos Models - View it on GitHub