Automatically split your PyTorch models on multiple GPUs for training & inference
Stars
656
Forks
44
Watchers
656
Open Issues
35
Overall repository health assessment
No package.json found
This might not be a Node.js project
29
commits
10
commits
3
commits
2
commits
Merge pull request #106 from BlackSamorez/sharding_refactor
a7d1939View on GitHubMerge pull request #113 from BlackSamorez/prepare_model_inputs
807a2ffView on GitHubMerge pull request #110 from BlackSamorez/pretrained_fix
c40153cView on GitHub