Thanks for the clarification; I think I’ve almost grasped the concept of batched matrix inversion and will submit a PR soon.