jax

e4b57cd1 - Add an example test for using `compute_on2` to multi-stream Pallas/Mosaic GPU kernels.