jax

f7337b26 - Fix bug where jnp.int64(1 << 40) would truncate even if explicit x64 types are allowed.