Joint Feature and Differentiable $ k $-NN Graph Learning using Dirichlet Energy | Read Paper on Bytez