在python中加速矩阵乘积B.T@A@B，其中B是稀疏的，A是完全密集的

Question

我想在python中加速以下矩阵乘积的计算：B.T@A@B，其中

A = 全密对称矩阵

B = 稀疏矩阵

这些矩阵的典型维度是 A = (40000, 40000), B = (40000, 2000)

如果使用 GPU 不是其中之一，我有哪些选择？