xla
Add test to train sharded resnet50 model
#4141
Merged

Commits
  • Add SPMD benchmarking test
    jonb377 committed 2 years ago
Loading