xla

52859ad8 - Added first sketch of model parallelism and fixed multi-device interaction between PT/XLA and autograd.