模型训练必备工具推荐:精选AI软件与开源项目

Trainer

Trainer

实验日志Github

基于PyTorch的通用模型训练框架

sd-scripts

sd-scripts

LoRAGithub

Stable Diffusion模型训练与生成的综合脚本工具库

mistral-finetune

mistral-finetune

Mistral-finetuneGithub

基于LoRA技术的Mistral模型高效微调框架

UniControl

UniControl

任务推理Github

优化多任务条件生成的统一扩散模型

OpenLRM

OpenLRM

模型训练开源项目

开源3D重建模型

YuzuMarker.FontDetection

YuzuMarker.FontDetection

Github开源项目

中日韩字体识别模型和场景文本图像数据集生成

fontogen

fontogen

字体推理Github

自定义字体生成工具

feast

feast

数据基础设施Github

旨在为机器学习平台团队提供可靠的特征管理工具的开源的特征存储系统

machinelearning

machinelearning

自定义模型Github

跨平台开源框架,简化.NET应用中的模型开发与部署

edward2

edward2

Github概率编程

概率编程语言,为深度学习生态系统设计,支持编写和操控模型用于灵活的训练和推断

TensorFlowASR

TensorFlowASR

GithubTFLite

TensorFlow 2中的智能自动语音识别解决方案

recommenders

recommenders

TensorFlow Recommenders开源项目

利用TensorFlow构建推荐系统模型的库

vocal-remover

vocal-remover

音源分离Github

使用深度学习的开源伴奏提取工具

lightning-flash

lightning-flash

lightning-flashGithub

跨数据领域和任务的AI模型训练与处理解决方案

ml-cvnets

ml-cvnets

Github开源项目

灵活的计算机视觉模型训练库

labml

labml

模型训练Github

通过手机监控深度学习模型训练和硬件使用情况

ffcv

ffcv

ffcvGithub

插入式数据加载系统,可显著提高模型训练中的数据吞吐量

LLamaTuner

LLamaTuner

Supervised fine-tuning datasetGithub

大语言模型微调工具,支持几乎所有GPU

llama3-Chinese-chat

llama3-Chinese-chat

中文化Github

llama3中文版上线,提供多样化微调和部署选项

StableSR

StableSR

开源项目图像超分辨率

通过扩散模型实现实际应用中的图像超分辨率