xla

3f54fa21 - Updating ml_framework_name and version to pytorch/xla2 in jax_configs. (#7211)