onnxruntime
ONNX Runtime 是微软开源的跨平台高性能推理与训练加速引擎,支持多种硬件后端,用于高效部署机器学习模型。
编辑点评作为工业级推理引擎,它兼容 PyTorch、TensorFlow 等导出的 ONNX 模型,并提供量化、图优化和异构执行能力。适合需要低延迟、高吞吐的模型服务场景,尤其在 Windows/Linux/边缘设备上表现稳定。注意:部分自定义算子可能需要额外适配,且训练加速功能相对较新,生产环境以推理为主。
这里汇总了代码库收录的「AI 基础设施」场景开源项目,按 Star 总量排序,适合按方向系统性地挑选与对比工具。
共收录 79 个项目 · 第 3 / 4 页
ONNX Runtime 是微软开源的跨平台高性能推理与训练加速引擎,支持多种硬件后端,用于高效部署机器学习模型。
编辑点评作为工业级推理引擎,它兼容 PyTorch、TensorFlow 等导出的 ONNX 模型,并提供量化、图优化和异构执行能力。适合需要低延迟、高吞吐的模型服务场景,尤其在 Windows/Linux/边缘设备上表现稳定。注意:部分自定义算子可能需要额外适配,且训练加速功能相对较新,生产环境以推理为主。
Hugging Face 推出的数据集中心库,提供海量即用数据集与高效数据操作工具,让 AI 模型的数据加载和预处理更简单、更快速。
编辑点评这是 AI 数据领域的核心基础设施,尤其适合需要快速获取和预处理标准数据集的开发者。它支持流式加载和内存映射,能处理超大规模数据;但与 Hugging Face 生态绑定较深,自定义格式需额外适配。无论是微调模型还是构建 RAG 应用,都能显著提升数据准备效率。
微软开源的推荐系统最佳实践库,提供构建、评估和部署推荐系统的完整解决方案与示例代码。
编辑点评这是推荐系统领域最全面的开源项目之一,涵盖从数据处理、模型训练到在线服务的完整流程,包含深度模型、经典算法和评估工具。适合希望快速上手推荐系统或寻找最佳实践的工程师与研究人员。注意项目体量较大,需按需选择模块,且部分功能依赖较重的运行环境。
基于云原生技术栈构建多模态 AI 应用的 Python 框架,提供从数据索引到推理部署的全流程支持。
编辑点评Jina Serve 把多模态 AI 应用开发与云原生基础设施深度结合,支持 gRPC、HTTP、流式通信和微服务编排,适合需要将视觉、文本、音频等模型封装为可扩展服务的工程团队。它的优势在于抽象程度高、部署灵活,但与成熟 web 框架相比学习曲线偏陡,生态相对小众。如果你在构建复杂多模态服务或神经搜索系统,非常值得关注。
ONNX 是开放的机器学习模型互操作标准,解决不同深度学习框架间模型交换与部署的兼容性问题。
编辑点评值得所有 AI 开发者关注,因为它是打破框架孤岛的关键一环。通过 ONNX,你可以将 PyTorch、TensorFlow 等模型统一导出为标准格式,便于跨框架推理、优化和部署。适合需要模型多端部署、迁移到推理引擎(如 ONNX Runtime)的场景。缺点是生态更新滞后于新算子,部分复杂模型需手动补齐算子或转换失败。
一个精选的机器学习生产环境开源工具列表,涵盖部署、监控、版本管理和扩展,帮助工程师构建可靠的ML系统。
编辑点评这是MLE/MLOps工程师的必备参考,按场景分类整理了大量生产级工具,从模型部署到可观测性一应俱全。优点在于覆盖面广、社区维护活跃,能快速找到合适的轮子;但作为awesome列表,不提供深度对比,需要自行评估工具适配性。适合用于技术选型调研或搭建ML基础设施时参考。
一本开放式的机器学习工程实战手册,系统总结大模型训练与调优的踩坑经验、性能优化方法和实用工具,帮助工程师少走弯路。
编辑点评这是由资深工程师维护的硬核工程指南,内容覆盖分布式训练、混合精度、GPU/TPU 调优、故障排查等真实场景,代码示例丰富,适合正在训练大模型或优化训练管线的工程师。优点是实践性强、紧跟前沿;缺点是内容庞大且更新快,需要按需查阅而非通读。
邱锡鹏《神经网络与深度学习》(蒲公英书)理论版 v2 与通识版的开源项目,提供完整书籍内容与配套资源,帮助学习者系统掌握深度学习基础。
编辑点评这是国内经典的深度学习教材开源项目,理论版覆盖数学推导与核心算法,通识版降低门槛,适合不同基础的学习者。内容权威、更新及时,但需注意阅读时结合实践,避免只看理论。适合作为课程配套教材或自学参考。
一个精选自然语言处理(NLP)资源列表,涵盖论文、工具、库和数据集,帮助开发者系统学习和应用NLP技术。
编辑点评这是NLP领域最经典的综合资源清单之一,覆盖从经典方法到前沿研究的广泛内容,适合初学者入门和研究者检索。优点在于分类清晰、社区维护活跃;缺点是链接较多,需自行筛选质量。适合想系统了解NLP生态、寻找特定工具或论文的开发者。
Jupyter 平台,涉及backtest、bitcoin、blockchain
编辑点评适用于AI/大模型、数据科学场景。使用 Jupyter 开发,涉及backtest、bitcoin、blockchain等技术方向。建议关注其社区活跃度和文档完善程度。
微软开源的深度学习工具包,用于构建、训练和部署神经网络模型,支持分布式训练与高性能推理。
编辑点评CNTK 是微软出品的工业级深度学习框架,曾在语音识别等领域表现优异,支持多机多卡分布式训练,性能强悍。但官方已停止维护,社区活跃度低,且 API 和生态远不如 PyTorch/TensorFlow。适合学习早期深度学习框架设计或维护遗留项目,新项目不建议选用。
Tensor2Tensor 是 Google 开源的深度学习模型与数据集库,提供标准化模块和丰富预置模型,旨在降低深度学习门槛并加速 ML 研究。
编辑点评作为 Transformer 等经典架构的原始实现库,Tensor2Tensor 对理解模型演进有重要参考价值。它内置大量数据集和训练流程,适合快速复现论文实验或进行教学演示。但项目已停止活跃维护,部分依赖较老,新项目建议直接使用 TensorFlow 或 PyTorch 生态的现代工具(如 Hugging Face)。
精选人工智能课程、书籍、视频讲座和论文的清单,帮助开发者系统学习AI基础知识与前沿技术。
编辑点评这是一个高质量AI学习资源聚合项目,内容覆盖从入门到进阶的课程、经典书籍、视频演讲和论文,适合想系统构建AI知识体系的开发者。优点是资源分类清晰、覆盖面广,且持续更新;缺点是需要自行筛选适合当前水平的内容。建议作为学习路径参考,按需深入阅读,而非从头到尾按顺序刷完。
将公式图片自动转换为 LaTeX 代码的 OCR 工具,基于 ViT 视觉 Transformer 模型,帮助用户快速复制数学公式。
编辑点评无需手动输入公式,拍照或截图即可生成 LaTeX,特别适合科研、教育场景中的文档写作。模型轻量且支持 CPU 推理,但复杂手写公式或低分辨率图片识别率会下降。适合需要频繁处理公式的 LaTeX 用户,也可作为 OCR 技术学习参考。
MNN 是阿里巴巴开源的高性能轻量级推理引擎,专注于端侧大模型和边缘 AI 场景,提供毫秒级延迟的深度学习模型部署能力。
编辑点评作为阿里生产级验证的推理引擎,MNN 在移动端和边缘设备上性能表现突出,支持 LLM 加速、算子丰富且多平台兼容。适合需要低延迟、低资源占用的端侧 AI 应用,但扩展新算子可能有一定学习成本,文档和社区生态仍在完善中。
Burn 是一个基于 Rust 的下一代张量库与深度学习框架,兼顾灵活性、高效性与可移植性,为 Rust 生态提供完整的模型训练与推理能力。
编辑点评Rust 深度学习框架稀缺,Burn 以纯 Rust 实现,内存安全且性能接近原生,支持多种后端(CPU/GPU)和动态图,适合追求高性能与可靠性的开发者。相比 PyTorch,其生态尚在成长,但设计现代,值得早期关注。
Unstructured 是一个开源 ETL 工具,可将复杂文档(如 PDF、Word)自动转换为干净、结构化的格式,便于大语言模型使用。
编辑点评它解决了非结构化数据预处理中的老大难问题,尤其适合 RAG 场景下的文档解析。支持多种格式、自带分区和清洗能力,能大幅减少手工处理成本。社区活跃,但部分高级功能依赖企业版,且对超复杂版式文档的识别精度仍需人工校验。适合需要快速搭建知识库或做文档分析的中小型团队。
Albumentations 是一个快速灵活的 Python 图像增强库,为深度学习模型训练提供丰富的数据增强操作,提升模型泛化能力。
编辑点评它是计算机视觉领域最流行的数据增强工具之一,基于 OpenCV 高效实现,支持多种主流深度学习框架。与 PyTorch、TensorFlow 无缝集成,适合图像分类、检测、分割等任务。丰富的增强策略和简洁的 API 让开发者能快速构建数据流水线,但在超大批量场景下需注意 CPU 瓶颈。
一个涵盖机器学习、深度学习、PostgreSQL、分布式系统、Node.js 和 Golang 的技术知识集合,帮助开发者系统性学习与进阶。
编辑点评内容覆盖面广,从 AI 到后端基础设施均有涉及,适合想拓宽技术栈的开发者作为方向参考。由于未提供具体文档结构和更新状态,建议按需查阅,勿当作系统教程。适合作为技术路线图或面试复习清单使用。
NVIDIA官方维护的深度学习示例集合,按模型分类,提供可复现的训练与部署脚本。
编辑点评这是NVIDIA精心整理的示例库,覆盖图像、语音、推荐等主流模型,代码质量高,针对GPU训练做了深度优化,适合在NVIDIA企业级基础设施上快速复现SOTA结果。优点是权威性和工程化程度高,示例完整;但资源消耗大,且强依赖NVIDIA硬件,入门门槛较高。适合已有一定深度学习基础、希望参考工业级实现或进行性能调优的开发者。