jax

34ce9f21 - Simplify implementation of _broadcast_to.

Simplify implementation of _broadcast_to. _broadcast_to needlessly squeezes away size 1 dimensions before passing its input to broadcast_in_dim. But broadcast_in_dim is perfectly happy to broadcast size 1 dimensions, so we don't need this squeeze.