Remove --jax_backend_target flag from main files.
With the new way of accessing TPUs presented at NeurIPS this is no longer necessary.
The main.py files are now nicely consistent. I don't think it makes sense to have a default value for the config.
PiperOrigin-RevId: 350131946
