How to load a checkpoint model with SHARDED_STATE_DICT?

How to load a checkpoint model with SHARDED_STATE_DICT?
I have a checkpoint which is place in a folder pytorch_model_0, which contains multiple distcp files.