xla

b64d8a2a - Add example to support pytorch lightning; misc bug fixes (#7054)