jax

d55be985 - [pallas] Pulled `default_memory_space` into `Mesh`

[pallas] Pulled `default_memory_space` into `Mesh` The default memory space is fixed for each mesh type, so it makes no sense to pass it to `default_mesh_discharge_rule` separately. PiperOrigin-RevId: 868718358