模型训练与微调
模型微调与训练工具
共 50 个项目
huggingface / diffusers
Hugging Face 推出的 PyTorch 扩散模型库，提供 SOTA 模型，轻松实现图像、视频、音频的生成与转换。
microsoft / onnxruntime
ONNX Runtime 是一个跨平台、高性能的机器学习推理与训练加速引擎，支持多种框架模型部署。
apache / airflow
Apache Airflow：以编程方式编写、调度和监控工作流的平台。
ggml-org / whisper.cpp
将 OpenAI Whisper 模型移植至 C/C++，实现无依赖、跨平台的高效本地语音转文字。
huggingface / transformers
统一框架，轻松加载、微调和部署文本、视觉、音频等多模态SOTA预训练模型。
ultralytics / yolov5
YOLOv5 是一个基于 PyTorch 的快速、准确的目标检测模型，支持导出至多种移动端和嵌入式格式。
deepspeedai / DeepSpeed
DeepSpeed 是深度学习优化库，让分布式训练和推理简单高效。
modelscope / FunASR
工业级语音识别工具包，支持超实时识别、50多种语言、说话人分离、情绪检测及流式服务。
sktime / sktime
统一的时间序列机器学习框架，集成预测、分类、异常检测等功能。
NVIDIA / cutlass
NVIDIA CUTLASS 提供 CUDA 模板和 Python DSL，用于高性能线性代数计算。
Project-MONAI / MONAI
面向医疗影像的深度学习AI工具包
nautechsystems / nautilus_trader
用Rust构建的生产级确定性事件驱动交易引擎，支持多资产与机器学习。
Automattic / mongoose
Mongoose 是 MongoDB 的 Node.js 对象建模工具，提供异步环境下的数据验证与封装。
Unstructured-IO / unstructured
将复杂文档（PDF、Word等）转换为结构化数据，为LLM提供干净输入的开源ETL工具。
lance-format / lance
面向多模态AI的开放湖仓格式，性能提升百倍，兼容主流数据工具。
tracel-ai / burn
Burn 是一个用 Rust 编写的新一代张量库与深度学习框架，兼顾灵活性、效率和可移植性。
deepchem / deepchem
DeepChem：民主化深度学习，赋能药物发现、量子化学、材料科学与生物学研究。
run-llama / llama_index
LlamaIndex 是一个用于构建 LLM 应用的文档处理平台，支持 RAG、智能体和多源数据集成。
skypilot-org / skypilot
SkyPilot将多云碎片化算力整合为一台超级计算机，助力AI团队快速构建前沿智能。
optuna / optuna
Optuna 是一个开源的超参数优化框架，能自动化、高效地搜索机器学习模型的最佳参数配置。
openbq-org / OpenBB
OpenBB是一个开源金融数据平台，为分析师、量化交易员和AI代理提供一站式市场数据与分析工具。
paperless-ngx / paperless-ngx
一个社区驱动的文档管理系统，支持扫描、索引和归档所有文档。
pyg-team / pytorch_geometric
PyTorch Geometric 是一个基于 PyTorch 的图神经网络库，提供高效的数据处理、丰富的模型和工具。
feast-dev / feast
Feast 是开源的特征存储平台，统一管理 AI/ML 特征，确保训练与推理数据的一致性。
snakers4 / silero-vad
Silero VAD 是一个预训练的企业级语音活动检测器，可精准识别音频中的语音片段。
dora-rs / dora
DORA 是一个面向数据流的机器人中间件，用 Rust 构建低延迟、可组合的分布式 AI 机器人应用。
NVIDIA / Megatron-LM
NVIDIA 开源的用于大规模训练 Transformer 模型的研究框架。
NVIDIA / warp
NVIDIA开源的Python框架，专为GPU加速的仿真、机器人及可微分机器学习打造。
online-ml / river
Python中的在线机器学习库，支持流式数据增量学习与概念漂移检测。
qdrant / qdrant
高性能、大规模向量数据库与搜索引擎，支撑下一代AI应用。
huggingface / pytorch-image-models
PyTorch 最大的图像编码器与骨干网络库，集成海量预训练权重及端到端脚本。
EleutherAI / lm-evaluation-harness
一个用于语言模型少样本评估的开源框架，支持多种任务和模型。
microsoft / qlib
微软开源的AI量化投资平台，集成多种机器学习模型和自动化研发工具。
invoke-ai / InvokeAI
InvokeAI是面向专业创作者的稳定扩散模型引擎，提供领先WebUI，支持图像生成与编辑。
autogluon / autogluon
AutoGluon 是一个自动化机器学习库，只需三行代码即可快速构建高精度模型，涵盖表格、文本、图像和时间序列数据。
explosion / spaCy
💫 Industrial-strength Natural Language Processing (NLP) in Python
volcano-sh / volcano
云原生批处理系统（CNCF 孵化项目），专为 AI、大数据、HPC 等场景设计。
intel / neural-compressor
SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime
unicode-org / icu4x
为客户端和资源受限环境提供轻量级、模块化的国际化 Rust 库。
deepinsight / insightface
InsightFace 是先进的人脸分析开源库，提供检测、对齐、识别及年龄估计等全套能力，支持多框架部署。
astroautomata / PySR
Symbolic Learning in Python and Julia
naptha / tesseract.js
纯 JavaScript 实现的 OCR 引擎，支持 100+ 语言，基于 WebAssembly 在浏览器和 Node.js 中运行。
speechbrain / speechbrain
A PyTorch-based Speech Toolkit
model-checking / kani
Kani 是 Rust 的模型检查器，通过穷举状态空间，在数学层面捕捉未定义行为与深层 Bug。
civitai / civitai
Stable Diffusion 模型与资源的社交分享平台
f / prompts.chat
开源AI提示词社区，支持分享、发现与收集，可自托管保障隐私。
evidentlyai / evidently
开源ML/LLM可观测性框架，提供100+指标，评估、测试和监控AI系统与数据管道。
modelscope / modelscope
ModelScope 是阿里巴巴达摩院推出的模型即服务平台，提供海量预训练模型，支持一键下载与部署。
voxel51 / fiftyone
一个用于可视化、分析和优化计算机视觉数据集的开源工具。
pymc-labs / pymc-marketing
Bayesian marketing toolbox in PyMC. Media Mix (MMM), customer lifetime value (CLV), buy-till-you-die (BTYD) models and more.