В настоящее время эта операция не поддерживается. Это трудно сделать параллельно и редко полностью необходимо. Вместо этого вы можете собрать nlargest, а затем вычислить pd.rank для вычисленного результата панд?
В качестве альтернативы, если вы пытаетесь вычислить ранг по столбцам, вы можете использовать map_partitions
df.map_partitions(pd.rank, axis="columns")