versae's picture
Add sharded pytorch
79cd49c