Efficient and Automatic 3D Parallelism Strategies Search via Contrastive Reinforcement Learning Pretrained Neural Networks.

Jie Ou, Xiaowang Li, Yueming Chen, Jiahong Qian, Teng Su, Wenhong Tian

Browse the full DSAA paper archive.