xla

d3ff030b - reimplement jax.image.scale_and_translate instead of jax.image.resize for _upsample_bilinear2d_aa (#8208)