lucadiliello
added tp sharded ckpts
8c8b767