机器学习框架排行榜

机器学习框架,按 GitHub 星标从高到低排列 · 共收录 10 款

训练与推理框架,面向开发者,需自行准备数据与算力。

1pytorchpytorch 104.9kPyTorch 是一个 Python 深度学习框架,提供带 GPU 加速的张量计算与基于动态图的自动求导,是科研与生产中训练神经网络的主流选择之一。Python 2026.09.302Lightning-AIpytorch-lightning 31.4kPyTorch Lightning 是一个深度学习训练框架(Python),把反向传播、混合精度、多 GPU 与分布式训练等工程代码自动化,让同一份模型代码从单卡扩展到上万卡而无需改动。Python 2026.09.103dmlcxgboost 28.9kXGBoost 是一个优化的分布式梯度提升库(C++ 核心,提供 Python、R、Java 等多语言绑定),以高效并行的树提升算法解决表格数据上的分类与回归问题,可运行到数十亿样本规模。C++ 2026.09.154microsoftonnxruntime 22.2k微软开源的跨平台机器学习推理和训练加速器,支持PyTorch、TensorFlow/Keras等深度学习框架及scikit-learn、LightGBM等经典机器学习库。C++ 2026.10.265lightgbm-orgLightGBM 18.8k一款基于决策树算法的高效、分布式梯度提升(GBDT)框架,用于排序、分类等机器学习任务。C++ 2026.07.186google-deepmindmujoco 15.8k一个由Google DeepMind维护的通用物理引擎,用于机器人、生物力学、图形和机器学习等领域的快速、精确仿真。C++ 2026.10.227quantumlibCirq 5.1kCirq 是 Google 开源的 Python 量子电路框架,用于创建、编辑与运行 NISQ 量子电路,可在内置模拟器或真实量子硬件上执行。Python 2026.06.308strawberry-graphqlstrawberry 4.7kStrawberry 是基于 Python 类型注解的 GraphQL 库,用 dataclasses 风格定义 schema,支持 Django、FastAPI、Starlette 等多种服务端集成。Python 2026.10.269benfredimplicit 3.8kimplicit 是一个面向隐式反馈数据的快速协同过滤推荐库(Python),提供 ALS、BPR、逻辑矩阵分解与物品近邻等算法,多线程 CPU 训练并支持 GPU 加速。Python 2026.05.0810determined-aidetermined 3.2k一个开源深度学习平台,简化分布式训练、超参数调优、实验追踪和资源管理,兼容PyTorch和TensorFlow。Go 2025.03.19