alpa-projects/alpa

Training and serving large-scale neural networks with auto parallelization.