在python中加快欧几里得距离

Question

我正在使用具有260GB RAM的32核计算机来执行以下操作：

# features is a numpy array of shape (2932480, 4096)

# query is a (4096,) - dimensional descriptor


ans = np.argsort(np.linalg.norm(features - query, axis=1))[:100]

计算欧几里德范数的总时间约为61秒，我可以通过哪些方式并行进行整个计算？