coremltools

收录理由

任何面向 iOS、macOS、watchOS 或 tvOS 开发机器学习功能的团队，迟早都要面对同一个问题：训练好的模型怎样才能在 Core ML 上跑起来。coremltools 正是苹果官方提供的 Python 工具，专门负责把 PyTorch、TensorFlow、scikit-learn、XGBoost 或 LibSVM 训练出的模型转换成设备端运行时所需的 .mlpackage 格式。除了转换，它还支持直接编辑和优化模型，比如通过量化压缩体积以适应移动设备的存储与内存限制，并且可以在 Mac 上先验证预测结果再随应用发布。作为官方管道，它也是通往 Neural Engine 加速的最直接路径，让推理在设备本地完成，既省电又无需联网，用户数据不必离开设备。对于在苹果硬件上维护大量模型的团队来说，其编辑原语也能在不需要重新训练的情况下，方便地调整或裁剪已转换的模型。