基于 PyTorch 的扩散模型工具箱,统一加载和训练 Stable Diffusion、FLUX 等模型,做图像、视频、音频生成。
开源的对话语音生成模型,能把文字转成自然流畅的日常对话语音,适合给聊天机器人和语音助手配音。
基于 PyTorch 的图神经网络库,提供图卷积等 GNN 模型的构建与训练组件,省去自己实现图数据结构和消息传递的麻烦。
基于 PyTorch 的深度强化学习库,提供 DQN、DDPG、A2C、模仿学习等算法实现,接口简洁,方便快速搭建和对比实验。
基于 PyTorch 的医学影像深度学习工具包,提供数据加载、预处理、网络结构和评估等组件,帮研究者快速搭建医疗影像 AI 流程。
基于 PyTorch 的度量学习库,提供对比损失、难例挖掘、采样器和检索指标,用来训练和评估相似度搜索所需的嵌入向量。
一套开箱即用的 2D/3D 人脸分析算法库,集成 ArcFace、RetinaFace 等模型,覆盖人脸检测、对齐、识别和年龄估计,支持 PyTorch、MXNet 等多后端。
预训练语音合成模型集合,支持俄语、亚美尼亚语等多种语言,几行代码就能把文本转成语音,省去自己训练模型的麻烦。
基于 PyTorch 的强化学习算法库,提供 PPO、SAC、TD3 等常用算法的可靠实现,开箱即用,适合机器人控制和仿真环境训练。
从下载数据到生成文本,手把手教你从零训练自己的大语言模型,适合想搞懂 LLM 训练全流程的开发者动手实践。
一个面向 NeRF 与高斯泼溅的协作式训练与可视化工作室,把多视角照片转成三维场景,简化了从数据到渲染的完整流程。
在 PaLM 架构上实现 RLHF 人类反馈强化学习,复现 ChatGPT 式训练流程,适合研究大模型对齐与强化学习微调。
面向 AI 开发者的实验追踪与模型管理平台,记录训练指标、超参数和数据集版本,支持从实验到生产的全流程协作。
医学影像分割的深度学习工具箱,开箱即用,能自动配置训练流程,在 CT、MRI 等 3D 数据上达到竞赛级精度。
提供一套稳健的训练配方,用人类和 AI 偏好数据对齐语言模型,涵盖 SFT、DPO 等流程,帮你把基座模型调成听话好用的助手。
用强化学习(GRPO)训练视觉语言模型做视觉理解,把 DeepSeek-R1 那套推理强化思路搬到多模态场景,适合研究 VLM 推理能力。
基于 PyTorch 的激光雷达 3D 目标检测工具箱,统一了 PointPillar、PV-RCNN 等主流点云检测模型的训练与评测流程,省去自己搭框架的麻烦。
开箱即用的视频扩散模型工具箱,支持文生视频和图生视频,重点解决高质量视频训练数据不足的问题。
在 Jupyter 里交互式可视化 BERT、GPT-2、RoBERTa 等 Transformer 模型的注意力权重,帮助理解模型内部机制。
自动检测和修复真实数据集中标签错误、异常样本等质量问题,帮你在训练前把脏数据清理干净,提升模型效果。
开源音乐生成基础模型,用符号规划、零样本翻唱和智能体编辑从歌词生成完整歌曲,适合研究音乐生成与风格迁移。
冰霜之地的技术博客合集,深入拆解 Go、HTTP/2、密码学、算法与机器学习等源码,适合想系统啃底层实现和系统设计的工程师。
提供文本嵌入与重排模型,用于语义检索和 RAG 场景,把文档和查询转成向量并做相关性排序。
开源文生视频和图生视频模型,CogVideoX 支持从文本或图片生成视频,适合搭建视频生成推理服务。