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