hummingbird
Hummingbird通过将训练好的传统机器学习模型编译为张量计算,使其能够借助神经网络框架(如PyTorch)加速。用户可利用神经网络框架的优化和硬件加速,无需重新设计模型。支持多种模型如决策树、随机森林、LightGBM和XGBoost,并提供易于替换的推理API。支持转换为PyTorch、TorchScript、ONNX和TVM格式。