模型训练

RaveForce 开源

用强化学习训练音乐生成模型，像玩Gym一样做实验

RaveForce 是一个面向音乐生成实验的 OpenAI Gym 风格工具包。它解决了音乐生成研究中缺乏标准化实···

neptune-client 开源

一键记录训练实验，自动追踪指标与模型，团队协作更高效

neptune-client 是 Neptune.ai 的官方 Python 客户端库，用于跟踪和记录基础模型训练过程中的实验数···

ml-workflow-automation 开源

在 Notebook 里一键把模型部署到 Kubernetes，实现 ML 工作流自动化。

这是一个用 Jupyter Notebook 演示典型机器学习工作流的 Python 项目，核心亮点是将模型自动部署为···

ludwig 开源

写个 YAML 就能训练大模型，告别繁琐代码

Ludwig 是一个基于 Python 的低代码机器学习框架，旨在让开发者无需编写大量底层代码即可构建、训练···

gokart 开源

让机器学习流水线可复现、可管理，告别混乱实验。

Gokart 是一个基于 Python 的机器学习流水线框架，旨在解决 ML 实验中的可复现性、任务依赖管理、代···

PaddleFormers 开源

在飞桨上跑通大模型微调与训练，开箱即用

PaddleFormers 是百度飞桨（PaddlePaddle）生态下的大模型训练与工具库，定位为易用的大语言模型预···

hypertune 开源

几行代码自动调参，快速找到最优模型参数

hypertune 是一个用于超参数优化的 Python 库，旨在简化机器学习模型的调参流程。它解决了手动网格···

catalyst 开源

让深度学习实验跑得更快、更规范，专注模型创新

Catalyst 是一个面向深度学习研究与开发的 Python 框架，旨在加速从实验设计到模型部署的全流程。它···

vae-lagging-encoder 开源

修复VAE后验坍缩，让生成模型更稳定多样

这是一个基于PyTorch的变分自编码器（VAE）改进实现，对应ICLR 2019论文《Lagging Inference Netwo···

nni 开源

模型调参交给机器自动试，最优结果坐等出炉

微软开源的 AutoML 工具包，覆盖机器学习生命周期的自动化，包括特征工程、神经网络架构搜索、模型···

higgsfield 开源

让万亿参数模型训练不再怕GPU故障，自动容错编排

higgsfield 是一个面向超大规模模型训练的容错型 GPU 编排与机器学习框架，专为数十亿至数万亿参数···

katib 开源

在K8s上自动调参，让模型优化省心省力

Katib 是一个基于 Kubernetes 的自动化机器学习（AutoML）平台，旨在解决在云原生环境中进行超参数···

sagify 开源

一条命令把本地模型部署到 AWS SageMaker，省心省力

sagify 是一个旨在简化大型语言模型和机器学习工作流的开源工具，主要面向 Python 开发者。它解决了···

PyTorch-CycleGAN 开源

用无配对图片训练，轻松实现风格迁移和图像转换

这是一个基于PyTorch的CycleGAN清洁可读实现，用于无配对图像到图像的转换。它解决了传统GAN需要成···

amazon-sagemaker-examples 开源

照着跑一遍，轻松上手 AWS 机器学习全流程

amazon-sagemaker-examples 是亚马逊官方维护的 Jupyter Notebook 示例集，旨在帮助开发者快速上手···

OpenSeq2Seq 开源

用配置文件快速搭建语音识别、翻译和语音合成模型，支持多卡混合精度训练。

OpenSeq2Seq 是 NVIDIA 开源的序列到序列学习工具包，专注于语音识别、文本转语音和自然语言处理任···

trainer 开源

在 Kubernetes 上轻松跑起分布式模型训练和微调

trainer 是一个基于 Kubernetes 的分布式 AI 模型训练与 LLM 微调平台，由开源社区维护。它解决了在···

evostra 开源

用 Python 快速实现进化策略，轻松训练强化学习模型

evostra 是一个用 Python 实现的快速进化策略（Evolution Strategy, ES）库。它解决了传统进化算法···

wandb 开源

训练过程实时看曲线，实验记录自动归档不丢

面向 AI 开发者的实验追踪与模型管理平台，支持训练过程的实时监控、指标可视化与团队协作，并能对···

benderopt 开源

无梯度也能调参，黑盒优化一键搞定

benderopt 是一个黑盒优化库，旨在帮助用户在没有梯度信息的情况下，对任意函数进行参数调优。它解···