xla

d127022f - Move JAX import into make_kernel_from_pallas and use _init_computation_client