GitHub repository: Erland366/alpa
Description: Training and serving large-scale neural networks with auto parallelization.