Add GPU & Distributed Training Support#22

Open

JoeHolt wants to merge 9 commits into

Commits on Apr 5, 2021

Copied over to new PR
JoeHolt
committed

Commits on Apr 9, 2021

Moved over test file
JoeHolt
committed

Commits on Apr 14, 2021

Updated dist to have new batch size API
JoeHolt
committed

Commits on Apr 16, 2021

Updated batch size setup
JoeHolt
committed

Commits on Apr 25, 2021

Commits on May 4, 2021

add example usage
stsievert
committed

Commits on May 5, 2021

tests pass
stsievert
committed
uncomment
stsievert
committed