PyTorch extensions for fast R&D prototyping and Kaggle farming
Stars
1.6k
Forks
126
Watchers
1.6k
Open Issues
6
Overall repository health assessment
No package.json found
This might not be a Node.js project
1.2k
commits
2
commits
2
commits
2
commits
1
commits
1
commits
1
commits
1
commits
Merge pull request #109 from liangosc/ndarray-should-be-array
91dc0fdView on GitHubMerge remote-tracking branch 'origin/develop' into develop
3336e88View on GitHubAdd device_id parameter to init_process_group for CUDA device configuration
89b1dd4View on GitHubRename wait_for_the_master to master_node_first in distributed.py
ec97142View on GitHubAdd keepdim parameter to geometric_mean and harmonic_mean functions
316360fView on GitHubReplace deprecated torch.cuda.amp.autocast with torch.amp.autocast
531822bView on GitHubvstack_header now can wrap text into multiple lines
47a1273View on GitHub