xla

ef6b3703 - [torch_ xla2] LLama3 example - fixed sharding configuration on LLama3 example (#7973)