I need to implement a method that computes the geometric median of a set of points. Given that this is part of a routine of a neural network model that runs on GPU (PyTorch