jax

52fa165d - Simplify promote_shapes.

Simplify promote_shapes. We can use lax.broadcast_to_rank instead of the considerably more complicated _broadcast_to. Add a fast path to broadcast_to_rank and broadcast to avoid emitting an equation if the rank is already correct.