TF top_k finds the largest so the input was made negative, Torch has an argument to find smallest instead of largest
TF top_k finds the largest so the input was made negative, Torch has an argument to find smallest instead of largest

deepchem/deepchem,deepchem/deepchem