This repo contains code for Unified-IO 2, including code to run a demo, do training, and do inference. This codebase is modified from T5X. [2/15/2024] We release the Pytorch code for unified-io 2.