Skip to content
Flax - Test

Rewritten flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate using jax.device_put #3391

Sign in to view logs

Rewritten flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate using jax.device_put

Rewritten flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate using jax.device_put #3391

Run Tests (3.12, mypy, newest)

Run Tests (3.12, mypy, newest)
succeeded Nov 19, 2025 in 1m 13s