xla

632153d0 - initial commit

initial commit Fix linters Fix comment Move JAX import into make_kernel_from_pallas and use _init_computation_client Guard JAX import for TPU tests only Fix linters Fix cpu/gpu ci