Implementation of TriMap dimensionality reduction in PyTorch - View it on GitHub