Roll-back https://github.com/google/jax/pull/14526 because it breaks view()
on scalar inputs
#14545
test_510276241% was force-pushed and no longer has any new commits.
Pushing new commits will allow the pull request to be re-opened.