模型训练

汇聚全球AI模型训练与微调项目,LoRA、全参微调、蒸馏、训练框架工具大全。

共收录 626 个开源项目

automix 开源 免费 仅开源

让多个AI模型互相验证,推理结果更靠谱

automix 是一个用于混合语言模型的开源项目,核心思路是通过自验证(self-verification)和元验证(···

★ 120 2023-10-09
AlignProp 开源 免费 仅开源

直接反向传播奖励,让扩散模型更听话,效率提升25倍

AlignProp 是一个用于对齐大规模文本到图像扩散模型的开源项目,核心思想是直接对奖励函数进行反向···

★ 330 2023-10-06
hf_ehr 开源 免费 仅开源

用 EHR 数据微调 HuggingFace 模型,快速构建医疗预测应用

hf_ehr 是一个专注于在电子健康记录(EHR)数据上训练 HuggingFace 模型的开源项目。它解决了医疗领···

★ 55 2023-10-05
makani 开源 免费 仅开源

让 AI 气象模型在数百 GPU 上并行训练,实验迭代从月缩到天。

Makani 是 NVIDIA 开源的天气与气候模型大规模并行训练框架,基于 Python 开发。它把原本只能在超算···

★ 405 2023-09-26
LongLoRA 开源 免费 仅开源

用少量算力把大模型上下文拉到 100k+,长文本任务轻松搞定

LongLoRA 是一个针对大语言模型的长上下文微调方案,由 MIT 等机构提出,相关论文入选 ICLR 2024 O···

★ 2685 进阶 · 约 30 分钟 · 本地安装 2023-09-21
aikit 开源 免费 仅开源

一条命令微调并部署开源大模型,省去繁琐工程配置。

aikit 是一个面向开源大语言模型的轻量级工具链,旨在简化 LLM 的微调、构建和部署流程。它解决了开···

★ 543 进阶 · 约 25 分钟 · 容器部署 2023-09-20
optimum-quanto 开源 免费 仅开源

几行代码把 PyTorch 模型压成 int8/int4,省显存提速

optimum-quanto 是 Hugging Face 生态下为 optimum 提供的 PyTorch 量化后端,目标是让模型在推理和···

★ 1056 入门 · 约 5 分钟 · 库/依赖 2023-09-19
nanotron 开源 免费 仅开源

极简代码跑通千亿模型3D并行训练

nanotron 是 Hugging Face 开源的大语言模型训练框架,核心目标是用极简代码实现 3D 并行训练(数据···

★ 2856 入门 · 约 10 分钟 · 本地安装 2023-09-11
Medusa 开源 免费 仅开源

给大模型加几个解码头,生成速度翻倍,不用改架构

Medusa是一个用于加速大语言模型生成的轻量级框架,核心思路是在原始模型基础上添加多个解码头(de···

★ 2774 进阶 · 约 30 分钟 · 库/依赖 2023-09-10
alignment-handbook 开源 免费 仅开源

一键复现主流 LLM 对齐实验,从 SFT 到 DPO 全流程开箱即用。

alignment-handbook 是一个用于对齐语言模型(LM)与人类及 AI 偏好的开源工具箱,由 Hugging Face···

★ 5711 进阶 · 约 30 分钟 · 本地安装 2023-08-25
InstructERC 开源 免费 仅开源

用指令微调让情感识别更灵活,一个模型搞定多种任务

InstructERC 是一个基于指令微调的开源情感识别模型项目,旨在通过自然语言指令的形式统一各类情感···

★ 151 2023-08-25
chatgpt-finetune-ui 开源 免费 已失效

拖拽上传数据,点点鼠标就能微调 ChatGPT 模型

这是一个基于 Python 和 Streamlit 构建的 ChatGPT 微调 Web 界面,专为 gpt-3.5-turbo 模型设计。···

★ 173 2023-08-23
LoadLoraWithTags 开源 免费 仅开源

一键管理 LoRA 触发词,自动补全,快速开关

LoadLoraWithTags 是一个 ComfyUI 的自定义节点,用于管理和加载 LoRA 模型及其触发词。它通过 JSO···

★ 90 2023-08-23
foundation-model-stack 开源 免费 仅开源

按需拼装的 PyTorch 基础模型训练与推理组件库

foundation-model-stack 是 IBM 开源的一套面向基础模型(大语言模型、多模态模型)的开发组件集合···

★ 234 2023-08-23
OmniQuant 开源 免费 仅开源

把大模型压到4比特,速度更快内存更省,精度还不掉链子。

OmniQuant 是一个针对大型语言模型(LLM)的量化技术,发表于 ICLR 2024 并被评为 spotlight。它旨···

★ 969 高级 · 约 60 分钟 · 本地 (Py/CUDA) 2023-08-22
mergekit 开源 免费 仅开源

拖拽式合并大模型,轻松融合多个模型能力

mergekit 是一个用于合并预训练大语言模型(LLM)的工具集。它解决了模型合并过程中繁琐的工程问题···

★ 7442 进阶 · 约 15 分钟 · 命令行工具 2023-08-21
DoLa 开源 免费 仅开源

不重训模型,对比层输出,让大模型少说瞎话。

DoLa 是一个用于提升大语言模型事实性的解码策略,来自同名论文的官方实现。它通过对比不同层级的输···

★ 563 高级 · 约 60 分钟 · 本地GPU 2023-08-21
mlflow_for_ml_dev 开源 免费 仅开源

跟着示例跑通 MLflow,搞定实验跟踪与模型管理

mlflow_for_ml_dev 是一个面向机器学习开发者的 MLflow 示例代码库,主要用 Jupyter Notebook 演示···

★ 241 2023-08-13
perfusion-pytorch 开源 免费 仅开源

用少量样本给文本生成模型打补丁,实现精准图像编辑

perfusion-pytorch 是 NVIDIA AI 提出的 Key-Locked Rank One Editing 方法的非官方 PyTorch 实现,···

★ 238 2023-08-06
ms-swift 开源 免费 仅开源

一条命令微调千种大模型,从训练到部署全搞定

ms-swift 是魔搭社区推出的国产大模型全流程训练与部署工具,旨在解决开发者微调大模型时面临的框架···

★ 16077 入门 · 约 10 分钟 · 命令行工具 2023-08-01
sapientml 开源 免费+付费 可打开

上传表格数据,自动生成并筛出可用建模流水线

SapientML 是一个面向表格数据的生成式 AutoML 工具,目标是把“从原始表格到可用模型”的流程自动···

★ 447 2023-07-31
OpenRLHF 开源 免费 仅开源

基于 Ray 的强化学习框架,让大规模 RLHF 训练又快又稳

OpenRLHF 是一个基于 Ray 的强化学习框架,专为大规模语言模型(LLM)的智能体训练而设计。它解决了···

★ 10147 进阶 · 约 30 分钟 · Docker 2023-07-30
LLM-Finetuning-Toolkit 开源 免费 仅开源

像写单元测试一样微调大模型,实验更稳更可控。

LLM-Finetuning-Toolkit 是一个面向开源大语言模型的微调、消融和单元测试工具包。它主要解决研究人···

★ 871 进阶 · 约 20 分钟 · 命令行安装 2023-07-24
LLMs-from-scratch 开源 免费 仅开源

手把手从零造出 ChatGPT,原理代码一次搞懂

手把手教你用 PyTorch 从零实现一个 ChatGPT 级别的 LLM,逐行讲解代码与原理,是深入理解 Transfo···

★ 108157 进阶 · 约 15 分钟 · 本地安装 2023-07-23