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

Trainer

Trainer

TrainerPyTorch

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

sd-scripts

sd-scripts

Stable Diffusion机器学习

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

mistral-finetune

mistral-finetune

Mistral-finetuneLoRA

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

UniControl

UniControl

UniControl控制生成

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

OpenLRM

OpenLRM

OpenLRM开源

开源3D重建模型

YuzuMarker.FontDetection

YuzuMarker.FontDetection

YuzuMarker.FontDetectionCJK字体识别

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

fontogen

fontogen

FontoGen字体生成

自定义字体生成工具

feast

feast

Feast特征存储

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

machinelearning

machinelearning

ML.NET机器学习

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

edward2

edward2

Edward2深度学习

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

TensorFlowASR

TensorFlowASR

TensorFlowASR自动语音识别

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

recommenders

recommenders

TensorFlow RecommendersKeras

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

vocal-remover

vocal-remover

vocal-remover深度学习

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

lightning-flash

lightning-flash

lightning-flashPyTorch

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

ml-cvnets

ml-cvnets

CVNets计算机视觉

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

labml

labml

LabML深度学习

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

ffcv

ffcv

ffcv模型训练

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

LLamaTuner

LLamaTuner

LLamaTuner大语言模型

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

llama3-Chinese-chat

llama3-Chinese-chat

llama3中文化

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

StableSR

StableSR

StableSR图像超分辨率

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