Skip to content

Modify clipping implementation to avoid jnp.moveaxis, which causes undesirable all-to-all's in distributed environments. #4629

Modify clipping implementation to avoid jnp.moveaxis, which causes undesirable all-to-all's in distributed environments.

Modify clipping implementation to avoid jnp.moveaxis, which causes undesirable all-to-all's in distributed environments. #4629

Annotations

2 warnings

Python 3.9 on ubuntu-latest jax=newest

succeeded Sep 19, 2024 in 12m 52s