Amazon Technologies, Inc.

Justin Chiu

Yida Wang

Mu Li

George Karypis

Trishul Amit Madhukar Chilimbi

Shuai Zheng

Zhen Zhang

A resource set which includes multiple servers with a respective plurality of training computing devices is identified for training a machine learning model. The resource set is subdivided into partition groups, such that each partition group can store a respective replica of state information of the model. The model is trained using the partition groups. The training comprises a multi-stage gathering of a portion of the state information at training computing devices of a particular partition group. Different types of communication channels between training computing devices are used in respective stages of the gathering, including inter-server communication channels in one stage and an intra-server communication channel during another stage. A trained version of the model is stored.

Growing community of inventors

Seattle, WA, United States of America

Justin Chiu