docs: @Override
      public Collection<String> wordsNearest(INDArray words, int top) {
          Counter<String> distances = new Counter<>();
  
          for (String s : vocabCache.words()) {
              INDArray otherVec = lookupTable.vector(s);
              double sim = Transforms.cosineSim(Transforms.unitVec(words.dup()), Transforms.unitVec(otherVec.dup()));
              distances.incrementCount(s, (float) sim);
          }
  
          distances.keepTopNElements(top);
          return distances.keySetSorted();
      }