jax

841254d9 - [pallas] `pl.core_map` now support `functools.partial`ed functions