Jeff Rasley
|
734d8991c8
Transformer kernel release (#242)
|
4 years ago |
Jeff Rasley
|
f8fa132512
Add back torch/torchvision requirements
|
4 years ago |
Jeff Rasley
|
7cf65d0e02
Remove explicit torch version requirement
|
4 years ago |
Shaden Smith
|
27d8385148
Use torch.cuda.device_count() (#126)
|
4 years ago |
kouml
|
bca23057de
Add requirements for cifar10 (#79)
|
4 years ago |
dependabot[bot]
|
0ca56d4b23
Bump tensorflow-gpu from 1.14.0 to 1.15.2 (#39)
|
4 years ago |
Shaden Smith
|
438aa01773
Enables NCCL backend in @distributed_test (#13)
|
4 years ago |
Shaden Smith
|
188f7d4e0b
Testing documentation (#10)
|
4 years ago |
yuxionghe
|
577295f5fa
Add files
|
4 years ago |