Training and serving large-scale neural networks with auto parallelization.
machine-learning deep-learning compiler distributed-computing high-performance-computing distributed-training jax alpa auto-parallelization llm
-
Updated
Dec 9, 2023 - Python