Official implementation of DynPartition: Automatic Optimal Pipeline Parallelism of Dynamic Neural Networks over Heterogeneous GPU Systems for Inference Tasks
machine-learning reinforcement-learning scheduling pytorch neural-networks model-parallelism treelstm pipeline-parallelism dynamic-neural-network dynpartition
-
Updated
May 5, 2023 - Python