jax

f18b9143 - Remove scalar float0 XLA representation.

Remove scalar float0 XLA representation. Some places in the code (e.g. lower_fun in xla.py) we need to go from an XLA shape to a ShapedArray and cannot do this if we drop shape information earlier.