jax

13a22f6c - [pallas] Deduped `get_interpreter_effects` between `core_map_p` and `pallas_call_p`