garg-aayush / model-parallelism

Model parallelism for NN architectures with skip connections (eg. ResNets, UNets)

Geek Repo

Github PK Tool