Skip to content
Back to pull request #1514

Add support for merging array groups across multiple devices, with uneven sharding between devices. #15703

Sign in to view logs

Add support for merging array groups across multiple devices, with uneven sharding between devices.

Add support for merging array groups across multiple devices, with uneven sharding between devices. #15703

Annotations

1 warning

build-checkpoint (Python 3.10, jax=0.4.34)

build-checkpoint (Python 3.10, jax=0.4.34)
succeeded Jan 17, 2025 in 1m 13s