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