基于 PyTorch 的图神经网络库,提供图卷积等 GNN 模型的构建与训练组件,省去自己实现图数据结构和消息传递的麻烦。
收录最全的 PyTorch 图像骨干网络合集,涵盖 ResNet、ViT、ConvNeXt 等主流模型,自带训练、评估、推理和导出脚本及预训练权重,省去自己搭网络和找权重的麻烦。
配套 PyTorch 深度学习课程的开源教程与代码,从零基础讲到能独立训练模型,适合边看边动手练。
一份 PyTorch 资源大全,把教程、论文、项目、社区等链接整理在一起,方便按图索骥找学习材料和参考实现。
PyTorch 版 CycleGAN 与 pix2pix 实现,提供图像到图像翻译的完整训练与测试流程,支持无配对和有配对数据转换。
用 PyTorch 从零手写实现 65 道大厂 ML/AI 面试真题,配 Jupyter Notebook、自动评分器和 MCP AI 导师,适合系统刷题备战机器学习面试。
基于 PyTorch 的医学图像分割非官方实现,支持 2D 和 3D 数据,方便快速搭建和训练分割模型。
用 5 秒语音样本克隆音色,实时合成任意文本的语音,基于 PyTorch/TensorFlow 的说话人编码与声码器流水线。
从公开研究文献出发,用第一性原理复现 Claude Mythos 架构的理论实现,供研究者拆解其注意力机制与模型设计思路。
基于 PyTorch 的开源语音工具包,覆盖语音识别、说话人识别与分离、语音增强等任务,提供预训练模型和训练流程,方便快速搭建语音系统。
开源语音识别工具包,覆盖训练、推理、流式 ASR、VAD、标点恢复和说话人分离全流程,并提供 OpenAI 兼容与 MCP 服务接口。
预训练的企业级语音活动检测(VAD)模型,能精准切分音频中的说话段与静音段,支持 ONNX 和 PyTorch 部署,适合语音识别、语音指令等场景的前置处理。
快速且省显存的精确注意力实现,支持训练和推理,能显著加速 Transformer 长序列计算。
开源实现的 CLIP 多模态模型,提供图像-文本对比学习训练与预训练权重,支持零样本图像分类和跨模态检索。
RWKV 是一种兼具 RNN 与 Transformer 优势的语言模型架构,支持像 GPT 一样并行训练,推理时线性时间、常量显存、无限上下文,无需 KV 缓存。
Rust 写的张量库和深度学习框架,兼顾灵活性与效率,支持自动微分、跨平台后端和内核融合,可训练和部署神经网络。
OpenAI Whisper 语音识别模型的 C/C++ 移植版,无需 Python 和 GPU 依赖,在普通 CPU 上就能跑语音转文字。
基于码本查找 Transformer 的盲人脸修复模型,能在低质量、模糊或退化的人脸图像上恢复清晰细节,适合老照片修复和人脸增强场景。
Open3D 是一个现代 3D 数据处理库,提供点云、网格、RGB-D 和 3D 可视化等算法与工具,支持 C++ 和 Python,可用于 3D 感知、重建与机器学习研究。
NVIDIA 的大规模 Transformer 训练框架,支持张量并行、流水线并行等模型并行策略,用来在成百上千张 GPU 上高效预训练大语言模型。