xla

f4a612cb - Training example for llama3; misc changes to make the training work. (#7194)