模型训练

distributed-ml-patterns 开源

照着搭，快速搞定分布式机器学习系统

这是一个与 Manning 出版社书籍《Distributed Machine Learning Patterns》配套的开源项目，由 Yua···

focal-frequency-loss 开源

频域聚焦损失，让生成图像细节更清晰

Focal Frequency Loss 是一个基于频域分析的图像重建与合成损失函数库，来自 ICCV 2021 论文。它解···

lightning-hydra-template 开源

开箱即用的深度学习实验框架，配置驱动，快速复现和扩展。

这是一个面向机器学习实验的工程化模板，基于 PyTorch Lightning 和 Hydra 构建。它解决了深度学习···

ml-engineering 开源

大模型训练推理踩坑指南，照着调优省半年时间

这是一本面向机器学习工程师的开放书籍，系统梳理了大模型训练与推理全流程的工程实践。它解决的核···

denoising_diffusion 开源

用PyTorch Lightning复现DDPM，快速跑通扩散模型训练与生成

这是一个非官方的PyTorch实现，复现了论文《Denoising Diffusion Probabilistic Models》（DDPM）中···

FedML 开源

一键跨云调度，轻松搞定分布式与联邦学习

FedML 是一个统一且可扩展的机器学习库，专注于大规模分布式训练、模型服务和联邦学习。它提供跨云···

neural-compressor 开源

一键压缩大模型，低比特量化提速省内存

neural-compressor 是英特尔开源的模型压缩工具库，专注于低比特量化与稀疏化技术，支持 INT8、FP8···

SuperPiano 开源

在 Colab 里用 Transformer 一键生成钢琴曲

SuperPiano 是一个基于 Jupyter Notebook 的开源项目，专注于音乐 AI 模型的训练与音乐生成。它利用···

ai8x-training 开源

在电脑上训练量化模型，一键部署到毫瓦级 AI 芯片

ai8x-training 是亚德诺半导体（ADI，原 Maxim）为 MAX78000 与 MAX78002 两款超低功耗边缘 AI 微控···

nncf 开源

压缩模型，让 OpenVINO 推理更快更省资源

NNCF（Neural Network Compression Framework）是 Intel 开源的神经网络压缩框架，专为提升 OpenVI···

determined 开源

一键拉起分布式训练，自动调参省心省力

Determined 是一个开源机器学习平台，旨在简化分布式训练、超参数调优、实验追踪和资源管理。它支持···

kedro-mlflow 开源

在 Kedro 管道中一键启用 MLflow 实验追踪与模型版本管理

kedro-mlflow 是一个 Kedro 插件，旨在将 MLflow 的机器学习生命周期管理能力无缝集成到 Kedro 数据···

ml-template-azure 开源

在 Azure 上快速搭建自动化机器学习流水线，省去从零配置的麻烦。

ml-template-azure 是一个面向 Azure 机器学习平台的自动化 MLOps 起步模板，旨在帮助团队快速搭建···

trl 开源

几行代码给大模型做 RLHF 对齐与偏好微调

TRL（Transformer Reinforcement Learning）是 Hugging Face 推出的开源库，专注用强化学习与偏好优···

pytorch-generative 开源

一键复现经典生成模型，快速上手图像生成实验

pytorch-generative 是一个基于 PyTorch 的生成模型工具库，旨在让生成建模变得简单易用。它实现了···

Sklearn-genetic-opt 开源

用遗传算法自动调参，告别网格搜索的低效

sklearn-genetic-opt 是一个面向 scikit-learn 的超参数优化与特征选择工具，采用进化算法（如遗传···

Torch-Pruning 开源

一键剪枝任意 PyTorch 模型，自动处理复杂依赖，压缩提速不降准。

Torch-Pruning 是一个基于 PyTorch 的结构化剪枝工具库，核心实现 CVPR 2023 论文 DepGraph（依赖图···

pycaret 开源

几行代码跑完建模全流程，小白也能交出好模型

Python 生态中的开源低代码 AutoML 平台，几行代码即可完成数据预处理、模型训练、调参与对比。4.0···

Matlab-GAN 开源

用 MATLAB 跑通多种 GAN，从入门到 CycleGAN 一步到位

Matlab-GAN 是一个基于 MATLAB 实现多种生成对抗网络（GAN）的开源项目，涵盖从基础 GAN 到 Pixel2···

Self-Supervised-Gans-Pytorch 开源

用旋转自监督让 GAN 训练更稳，生成图像更逼真

Self-Supervised-Gans-Pytorch 是 CVPR 2019 论文《Self-Supervised GANs via Auxiliary Rotation ···

autogluon 开源

三行代码搞定高精度模型，自动调参集成，省时省力

AutoGluon 是一个自动化机器学习（AutoML）库，旨在用三行代码实现快速且准确的模型训练。它解决了···

aim 开源

几行代码接入，秒级对比海量实验，训练调参效率翻倍

Aim 是一个开源、易用的实验追踪工具，专为机器学习项目设计。它解决了 ML 团队在训练过程中难以高···

Megatron-LM 开源

NVIDIA 出品，高效训练千亿级大模型

Megatron-LM 是 NVIDIA 主导的深度学习研究项目，专注于大规模 Transformer 模型的训练。它解决了在···

GAN-XML-Fixer 开源

用 AI 自动修复损坏的 XML 文件，省去手工改错时间

GAN-XML-Fixer 是一个基于生成对抗网络（GAN）的 XML 文件自动修复工具。它针对 XML 文件因格式错误···