Skip to content
Lint

[JAX] Use TE with_sharding_constraint wrapper even if flax returns successful logical sharding rules #18482

Sign in to view logs

[JAX] Use TE with_sharding_constraint wrapper even if flax returns successful logical sharding rules

[JAX] Use TE with_sharding_constraint wrapper even if flax returns successful logical sharding rules #18482

Run time

Learn about OS pricing on GitHub Actions
Job Run time
PyTorch Python
2m 6s
PyTorch C++
41s
JAX C++
33s
JAX Python
41s
4m 1s