Skip to content
Back to pull request #5101

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

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 #3596

Check commit count

Check commit count
succeeded Dec 18, 2025 in 6s