DeepSpeed

8c5967f4 - add the load-checkpoint support for falcon-40b (reducing loading time from over 10 minutes to about 15 sec)