xla

74922eca - add several examples testing the failure modes of tensor parallel