Add a sharding rule for reduce_precision_p
and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn
's
#70418
Annotations
1 warning
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
|
Loading