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