模型训练

Awesome-Model-Quantization 开源

一页看全模型量化论文、工具与部署方案

这是一个专注于模型量化领域的精选资源合集，系统整理了论文、基准测试、综述和工具。量化是压缩神···

catalyst 开源

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

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

vae-lagging-encoder 开源

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

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

brevitas 开源

PyTorch 模型一键量化，低比特也能高精度跑上 FPGA

Brevitas 是 Xilinx（AMD）开源的 PyTorch 神经网络量化库，专注为 FPGA 与硬件加速场景提供从训练···

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