jax

ef7df1ae - [pallas_mgpu] Allow trees (eg tuples) to be returned from cond_p expressions.