jax

967391f0 - Remove fancy batching rule for lax.empty_p

Remove fancy batching rule for lax.empty_p This turned out to fail when in_axes=None. It is cleaner here to rely on the default batching behavior.