xgboost-metal 开源:XGBoost 移植到 Metal,Mac 可用 Apple GPU 训练梯度提升

HankYeomans · x · 2026-11-04

xgboost-metal 项目将 XGBoost 移植到 Apple 的 Metal,使梯度提升模型能充分利用 Apple Silicon(尤其是 Apple GPU)加速。它保留熟悉的 XGBoost Python API,可在 Mac GPU 上直接训练、预测、评估模型并计算特征贡献,无需更换工具链。

项目由原项目作者与 Bing Xu、int21ai 等合作完成,已在 GitHub 开源,对在 Mac 上做表格数据机器学习的开发者是实用的本地加速方案。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →