xgboost是否将TPU用于“ gpu-hist”（如果TPU可用）？

Question

我很好奇xgboost是否会在Google colab中使用TPU（如果可用）？当然，它可以很好地利用GPU ...

Answer 1

感谢您的提问。 TPU通常非常适合基于神经网络的模型。我还没有听说过在TPU上进行基于树的计算的方法，而且我非常怀疑TPU会在今天的情况下保持高性能。