Implement XLAShardedTensor.redistribute and test #1771
qihqi
This run and associated checks have been archived and are scheduled for deletion.
Learn more about checks retention
torchax.yml
on: pull_request
Check Code Changes
/
check_files
17s
Matrix: torchax-cpu