don't initialize backend in xla_computation
don't initialize backend in xla_computation

This should allow us to use donate_argnums *and* build HLO computations
for backends not available at build time.

tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax,google/jax