vamp-ire-tap

thanks a lot for the swift response, the `x` and `cond` variables do not seem to have nans: ``` (Pdb) !np.argwhere(np.isnan(x.cpu().detach().numpy())) array([], shape=(0, 3), dtype=int64) (Pdb) !np.argwhere(np.isnan(cond.cpu().detach().numpy())) array([], shape=(0, 3),...

thank you for your answer, but the torch.sort() is differentiable right? not easy to optimize, but differentiable nevertheless?

thank you!