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

vall-e

vall-e

VALL-EPyTorch

开源PyTorch框架下基于EnCodec Tokenizer的音频合成实现

ChatGPT_DAN

ChatGPT_DAN

ChatGPTDAN模式

详述ChatGPT DAN模式如何拓展人工智能应用边界

kogpt

kogpt

KoGPTKakaoBrain

韩语生成预训练模型

MotionGPT

MotionGPT

MotionGPT人机交互

构建多任务的人体动作与语言统一模型

zero_nlp

zero_nlp

zero_nlp模型训练

中文NLP训练与应用框架

safe-rlhf

safe-rlhf

Beaver安全RLHF

北京大学开发的AI安全增强框架

Awesome-LLM-Inference

Awesome-LLM-Inference

Awesome-LLM-Inference模型训练

一系列关于涵盖了从基础框架到先进技术的大型语言模型推理的研究论文和配套代码

pytorch-cifar100

pytorch-cifar100

Pytorchcifar100

使用PyTorch高效训练和测试CIFAR-100数据集

x-transformers

x-transformers

x-transformerstransformer

轻量级Transformer模型,支持完整的编解码器配置和最新研究成果,适合各种从图像分类到语言模型的应用

U-2-Net

U-2-Net

U2-Net人像分割

深度嵌套U结构助力显著对象精准检测

wandb

wandb

Weights & Biases机器学习

专注于机器学习流程优化的工具

tensor2tensor

tensor2tensor

Tensor2Tensor深度学习

一个旨在使深度学习更加深入的深度学习模型和数据集的库

gpt-neo

gpt-neo

GPT-Neo模型训练

使用 mesh-tensorflow 库的模型并行 GPT-2 和 GPT-3 样式模型的实现

DataDreamer

DataDreamer

DataDreamer合成数据生成

生成合成数据和优化模型的开源Python工具

pytorch-metric-learning

pytorch-metric-learning

PyTorch Metric LearningTripletMarginLoss

高级度量学习工具库,适用于深度学习

onnxruntime

onnxruntime

ONNX Runtime机器学习

跨平台的机器学习模型推理与训练加速工具

Paddle

Paddle

PaddlePaddle深度学习

拥有先进技术和丰富功能的工业平台

CLIP

CLIP

CLIPPyTorch

CLIP是一种在各种(图像、文本)对上训练的神经网络

BLOOM

BLOOM

AI开发模型训练

开发的多语言、大规模开放科学语言模型

网易云课堂

网易云课堂

AI图像图像生成

云端AI绘画和设计创作平台