Awesome-pytorch-list
Awesome-pytorch-list 是一个专为 PyTorch 开发者打造的开源资源导航库,旨在解决深度学习领域资源分散、难以检索的痛点。它系统性地整理了 GitHub 上成千上万个与 PyTorch 相关的高质量项目,涵盖从基础库、辅助工具到前沿模型实现的方方面面。
无论是自然语言处理(如机器翻译、语音合成)、计算机视觉,还是概率生成模型,用户都能在这里找到对应的成熟框架与代码示例,例如 AllenNLP、Fairseq 以及各类经典论文的复现版本。此外,它还收录了丰富的教程、书籍和技术会议资料,帮助使用者快速上手或深入钻研。
这份清单特别适合 AI 研究人员、算法工程师以及正在学习深度学习的学生使用。对于希望避免重复造轮子、快速寻找可靠代码基底的开发者而言,Awesome-pytorch-list 提供了极高的参考价值。其核心亮点在于分类清晰、更新及时且社区活跃,将零散的生态资源汇聚成一张详尽的“地图”,让用户能高效定位所需工具,显著提升研发与学习效率。
使用场景
某初创公司的算法工程师团队正致力于开发一款支持多方言的智能客服语音系统,急需在 PyTorch 框架下快速集成先进的语音识别与合成模块。
没有 Awesome-pytorch-list 时
- 资源检索如大海捞针:工程师需在 GitHub 海量仓库中盲目搜索,难以区分哪些是过时的 Demo,哪些是维护良好的生产级代码(如区分基础的
pytorch-seq2seq与成熟的fairseq-py)。 - 技术选型试错成本高:缺乏权威对比,团队可能误选不再维护的库(如旧版 NLP 工具),导致后期重构浪费数周时间。
- 前沿论文复现困难:找不到 SOTA(最先进)模型的具体实现地址,例如难以快速定位到
Tacotron-pytorch或espnet等端到端语音处理工具箱。 - 生态组件分散:需要分别寻找数据预处理、模型解释性(如 Captum)及多语言嵌入(如 MUSE)等辅助库,协作效率低下。
使用 Awesome-pytorch-list 后
- 一站式精准导航:直接通过分类目录锁定高质量资源,迅速选定
espnet作为核心语音引擎,并利用pyannote-audio解决说话人分离难题。 - 规避废弃项目风险:依据列表中的星标数和社区热度,直接排除过时方案,确保基于
AllenNLP或Hugging Face生态构建稳健基线。 - 加速论文落地:通过"Paper implementations"板块快速找到神经机器翻译和语音合成的官方复现代码,将研发周期从数月缩短至数周。
- 全链路工具整合:按图索骥集齐从音频 I/O (
pytorch/audio) 到模型可解释性的完整工具链,大幅提升系统迭代速度。
Awesome-pytorch-list 将原本碎片化的 PyTorch 生态整合为一张清晰的“寻宝图”,让开发者从繁琐的搜寻工作中解放出来,专注于核心算法的创新与落地。
运行环境要求
未说明
未说明

快速开始
超棒的 PyTorch 列表

目录
PyTorch 及相关库
NLP 和语音处理:
- pytorch text: 与 Torch 文本相关的资源。
- pytorch-seq2seq: 一个基于 PyTorch 实现的序列到序列(seq2seq)模型框架。
- anuvada: 使用 PyTorch 构建的可解释 NLP 模型。
- audio: 简单的 PyTorch 音频输入输出工具。
- loop: 一种跨多个说话人生成语音的方法。
- fairseq-py: Facebook AI Research 用 Python 编写的序列到序列工具包。
- speech: PyTorch 的自动语音识别实现。
- OpenNMT-py: 开源的 PyTorch 神经机器翻译系统 http://opennmt.net
- neuralcoref: 基于神经网络和 spaCy 的最先进共指消解模型 huggingface.co/coref
- sentiment-discovery: 大规模无监督语言建模,用于鲁棒的情感分类。
- MUSE: 一个多语言无监督或监督词嵌入库。
- nmtpytorch: 基于 PyTorch 的神经机器翻译框架。
- pytorch-wavenet: WaveNet 的快速生成实现。
- Tacotron-pytorch: Tacotron:迈向端到端语音合成。
- AllenNLP: 一个开源的 NLP 研究库,基于 PyTorch 构建。
- PyTorch-NLP: 适用于 PyTorch 的文本工具和数据集 pytorchnlp.readthedocs.io
- quick-nlp: 基于 FastAI 的 PyTorch NLP 库。
- TTS: 用于文本转语音的深度学习技术。
- LASER: 语言无关的句子表示方法。
- pyannote-audio: 用于说话人日志的神经网络构建模块:语音活动检测、说话人变化检测、说话人嵌入。
- gensen: 通过大规模多任务学习,学习通用分布式句子表示。
- translate: Translate——一个 PyTorch 语言库。
- espnet: 端到端语音处理工具箱 espnet.github.io/espnet
- pythia: 一套用于视觉问答的软件工具。
- UnsupervisedMT: 基于短语和神经网络的无监督机器翻译。
- jiant: Jiant 句子表示学习工具箱。
- BERT-PyTorch: Google AI 2018 年 BERT 的 PyTorch 实现,并附有简单注释。
- InferSent: 句子嵌入(InferSent)及用于自然语言推理的训练代码。
- uis-rnn: 这是无界交错状态循环神经网络(UIS-RNN)算法的库,对应于论文《完全监督的说话人日志》。arxiv.org/abs/1810.04719
- flair: 一个非常简单的框架,用于最先进的自然语言处理(NLP)。
- pytext: 一个基于 PyTorch 的自然语言建模框架 fb.me/pytextdocs
- voicefilter: Google AI VoiceFilter 系统的非官方 PyTorch 实现 http://swpark.me/voicefilter
- BERT-NER: 基于 BERT 的 PyTorch 命名实体识别。
- transfer-nlp: 专为灵活的研究和开发设计的 NLP 库。
- texar-pytorch: 用于机器学习和文本生成的工具包,基于 PyTorch texar.io
- pytorch-kaldi: pytorch-kaldi 是一个用于开发最先进的 DNN/RNN 混合语音识别系统的项目。其中 DNN 部分由 PyTorch 负责,而特征提取、标签计算和解码则由 Kaldi 工具箱完成。
- NeMo: 神经模块:用于对话式 AI 的工具包 nvidia.github.io/NeMo
- pytorch-struct: 核心结构化预测算法的向量化实现库(HMM、依存句法树、CKY 等)。
- espresso: Espresso:一个快速的端到端神经语音识别工具。
- transformers: Hugging Face Transformers:面向 TensorFlow 2.0 和 PyTorch 的最先进的自然语言处理工具。huggingface.co/transformers
- reformer-pytorch: Reformer,高效的 Transformer,在 PyTorch 中实现。
- torch-metrics: 用于 PyTorch 模型评估的指标。
- speechbrain: SpeechBrain 是一个基于 PyTorch 的开源、一体化语音工具箱。
- Backprop: Backprop 让使用、微调和部署最先进的机器学习模型变得简单。
简历:
- pytorch vision:专用于计算机视觉的数据集、变换和模型。
- pt-styletransfer:基于 PyTorch 的类实现的神经风格迁移。
- OpenFacePytorch:用于使用 OpenFace 的 nn4.small2.v1.t7 模型的 PyTorch 模块。
- img_classification_pk_pytorch:快速将你的图像分类模型与当前最先进的模型(如 DenseNet、ResNet 等)进行比较。
- SparseConvNet:子流形稀疏卷积网络。
- Convolution_LSTM_pytorch:一个多层卷积 LSTM 模块。
- face-alignment::fire: 基于 PyTorch 构建的 2D 和 3D 面部对齐库,adrianbulat.com。
- pytorch-semantic-segmentation:用于语义分割的 PyTorch 实现。
- RoIAlign.pytorch:这是 RoIAlign 的 PyTorch 版本。该实现基于 crop_and_resize,支持 CPU 和 GPU 上的前向和反向传播。
- pytorch-cnn-finetune:使用 PyTorch 微调预训练的卷积神经网络。
- detectorch:Detectorch——面向 PyTorch 的 detectron。
- Augmentor:用于机器学习的 Python 图像增强库。http://augmentor.readthedocs.io
- s2cnn: 该库包含用于球面信号(例如全向相机、地球表面信号)的 SO(3) 等变 CNN 的 PyTorch 实现。
- TorchCV:一个基于 PyTorch 的计算机视觉深度学习框架。
- maskrcnn-benchmark:在 PyTorch 中快速、模块化的实例分割和目标检测算法参考实现。
- image-classification-mobile:在 ImageNet-1K 上预训练的分类模型集合。
- medicaltorch:一个用于 PyTorch 的医学影像框架 http://medicaltorch.readthedocs.io
- albumentations:快速的图像增强库。
- kornia:可微分的计算机视觉库。
- pytorch-text-recognition:文本识别组合——CRAFT + CRNN。
- facenet-pytorch:从 davidsandberg/facenet 移植而来的预训练 PyTorch 人脸检测和识别模型。
- detectron2:Detectron2 是 FAIR 的下一代目标检测和分割研究平台。
- vedaseg:一个基于 PyTorch 的语义分割框架。
- ClassyVision:一个端到端的 PyTorch 框架,用于图像和视频分类。
- detecto:用不到 10 行代码即可实现的 Python 计算机视觉。
- pytorch3d:PyTorch3D 是 FAIR 提供的用于处理 3D 数据深度学习的可重用组件库 pytorch3d.org。
- MMDetection:MMDetection 是一个开源的目标检测工具箱,属于 OpenMMLab 项目。
- neural-dream:DeepDream 算法的 PyTorch 实现。能够生成梦幻般的幻觉视觉效果。
- FlashTorch:用于 PyTorch 神经网络的可视化工具包!
- Lucent:TensorFlow 和 OpenAI Clarity 的 Lucid 被适配到 PyTorch 上。
- MMDetection3D:MMDetection3D 是 OpenMMLab 下一代通用 3D 目标检测平台,属于 OpenMMLab 项目。
- MMSegmentation:MMSegmentation 是语义分割工具箱和基准测试平台,属于 OpenMMLab 项目。
- MMEditing:MMEditing 是图像和视频编辑工具箱,属于 OpenMMLab 项目。
- MMAction2:MMAction2 是 OpenMMLab 下一代动作理解工具箱和基准测试平台,属于 OpenMMLab 项目。
- MMPose:MMPose 是姿态估计工具箱和基准测试平台,属于 OpenMMLab 项目。
- lightly:Lightly 是一个用于自监督学习的计算机视觉框架。
- RoMa:一个轻量级且高效的库,用于处理 3D 旋转。
概率/生成式库:
- ptstat:在 PyTorch 中进行概率编程和统计推断。
- pyro:使用 Python 和 PyTorch 进行深度通用概率编程 http://pyro.ai
- probtorch:Probabilistic Torch 是一个扩展 PyTorch 的深度生成模型库。
- paysage:在 Python/PyTorch 中进行无监督学习和生成模型。
- pyvarinf:一个 Python 包,用于促进在 PyTorch 中使用基于变分推断的贝叶斯深度学习方法。
- pyprob:一个基于 PyTorch 的概率编程和推理编译库。
- mia:一个用于对机器学习模型发起成员身份推断攻击的库。
- pro_gan_pytorch:以 PyTorch nn.Module 扩展形式实现的 ProGAN 包。
- botorch:在 PyTorch 中进行贝叶斯优化。
其他库:
- pytorch extras:PyTorch 的一些额外功能。
- functional zoo:与 Lua Torch 不同,PyTorch 的核心内置了自动微分机制,因此无需依赖
torch.nn模块的模块化结构,可以直接创建所需的变量并编写函数来操作它们,这种方式有时更为便捷。该仓库以函数式方式定义模型,并为部分模型提供了预训练权重。 - torch-sampling:该包提供了一组用于从内存或磁盘数据中进行采样的变换和数据结构。
- torchcraft-py:TorchCraft 的 Python 封装,是连接 Torch 和 StarCraft 的桥梁,专为 AI 研究设计。
- aorun:Aorun 旨在成为以 PyTorch 为后端的 Keras。
- logger:一个用于实验的简单日志记录工具。
- PyTorch-docset:PyTorch 文档集!可与 Dash、Zeal、Velocity 或 LovelyDocs 配合使用。
- convert_torch_to_pytorch:将 Torch t7 模型转换为 PyTorch 模型及源代码。
- pretrained-models.pytorch:该项目的目标是帮助复现研究论文的结果。
- pytorch_fft:PyTorch 的 FFT 封装库。
- caffe_to_torch_to_pytorch:用于在 Caffe、Torch 和 PyTorch 之间进行模型转换的工具。
- pytorch-extension:这是一个基于 CUDA 的 PyTorch 扩展,用于计算两个张量的哈达玛积。
- tensorboard-pytorch:该模块以 TensorBoard 格式保存 PyTorch 张量,便于检查。目前支持 TensorBoard 中的标量、图像、音频和直方图类型数据。
- gpytorch:GPyTorch 是一个基于 PyTorch 实现的高斯过程库,旨在轻松构建灵活且模块化的高斯过程模型,即使非专家也能使用。
- spotlight:基于 PyTorch 的深度推荐模型。
- pytorch-cns:使用 PyTorch 进行压缩网络搜索。
- pyinn:结合 CuPy 的 PyTorch 神经网络融合操作。
- inferno:围绕 PyTorch 的实用工具库。
- pytorch-fitmodule:适用于 PyTorch 模块的超简单拟合方法。
- inferno-sklearn:一个兼容 scikit-learn 的神经网络库,封装了 PyTorch。
- pytorch-caffe-darknet-convert:在 PyTorch、Caffe prototxt/权重以及 Darknet cfg/权重之间进行转换。
- pytorch2caffe:将 PyTorch 模型转换为 Caffe 模型。
- pytorch-tools:PyTorch 相关工具集合。
- sru:以 CNN 的速度训练 RNN(arxiv.org/abs/1709.02755)。
- torch2coreml:将 Torch7 转换为 CoreML 格式。
- PyTorch-Encoding:PyTorch 深度纹理编码网络 http://hangzh.com/PyTorch-Encoding。
- pytorch-ctc:PyTorch-CTC 是一种基于 Beam Search 的 CTC(联结时序分类)解码实现,适用于 PyTorch。其 C++ 代码大量借鉴自 TensorFlow,并进行了改进以提高灵活性。
- candlegp:基于 PyTorch 的高斯过程实现。
- dpwa:通过成对平均法进行分布式学习。
- dni-pytorch:利用合成梯度为 PyTorch 提供解耦的神经接口。
- skorch:一个兼容 scikit-learn 的神经网络库,封装了 PyTorch。
- ignite:Ignite 是一个高级库,用于辅助在 PyTorch 中训练神经网络。
- Arnold:Arnold——DOOM 智能体。
- pytorch-mcn:将 MatConvNet 模型转换为 PyTorch 格式。
- simple-faster-rcnn-pytorch:简化版 Faster R-CNN 实现,性能具有竞争力。
- generative_zoo:该仓库提供了几种生成模型在 PyTorch 中的可用实现。
- pytorchviz:一个用于可视化 PyTorch 执行图的小型工具包。
- cogitare:Cogitare——一个现代、快速且模块化的 Python 深度学习和机器学习框架。
- pydlt:基于 PyTorch 的深度学习工具箱。
- semi-supervised-pytorch:在 PyTorch 中实现多种基于 VAE 的半监督和生成模型。
- pytorch_cluster:优化的图聚类算法 PyTorch 扩展库。
- neural-assembly-compiler:基于自适应神经编译的 PyTorch 神经汇编编译器。
- caffemodel2pytorch:将 Caffe 模型转换为 PyTorch 格式。
- extension-cpp:PyTorch 中的 C++ 扩展。
- pytoune:一个类似 Keras 的框架及 PyTorch 工具集。
- jetson-reinforcement:面向 NVIDIA Jetson TX1/TX2 的深度强化学习库,结合 PyTorch、OpenAI Gym 和 Gazebo 机器人仿真器。
- matchbox:以单个样本级别编写 PyTorch 代码,然后高效地在小批量上运行。
- torch-two-sample:用于双样本检验的 PyTorch 库。
- pytorch-summary:类似于 Keras 中
model.summary()的 PyTorch 模型摘要。 - mpl.pytorch:MaxPoolingLoss 的 PyTorch 实现。
- scVI-dev:scVI 项目的 PyTorch 开发分支。
- apex:一个实验性的 PyTorch 扩展(未来将被弃用)。
- ELF:ELF——一个用于游戏研究的平台。
- Torchlite:一个基于 PyTorch 的高级库。
- joint-vae:JointVAE 的 PyTorch 实现,用于分离连续和离散的变化因子。
- SLM-Lab:基于 PyTorch 的模块化深度强化学习框架。
- bindsnet:一个用于在 CPU 或 GPU 上使用 PyTorch 模拟脉冲神经网络 (SNN) 的 Python 包。
- pro_gan_pytorch:作为 PyTorch nn.Module 扩展实现的 ProGAN 包。
- pytorch_geometric:PyTorch 的几何深度学习扩展库。
- torchplus:在 PyTorch 模块上实现
+运算符,返回序列。 - lagom:lagom——一个轻量级 PyTorch 基础设施,用于快速原型化强化学习算法。
- torchbearer:一款面向研究人员的 PyTorch 模型训练库。
- pytorch-maml-rl:基于 PyTorch 的模型无关元学习强化学习。
- NALU:Trask 等人发表的“神经算术逻辑单元”论文中 NAC/NALU 的基础 PyTorch 实现 arxiv.org/pdf/1808.00508.pdf。
- QuCumber:神经网络多体波函数重构。
- magnet:自我构建的深度学习项目 http://magnet-dl.readthedocs.io/。
- opencv_transforms:Torchvision 图像增强的 OpenCV 实现。
- fastai:fast.ai 深度学习库、课程和教程。
- pytorch-dense-correspondence:用于“密集对象网络:为机器人操作而学习密集视觉对象描述符”的代码 arxiv.org/pdf/1806.08756.pdf。
- colorization-pytorch:Interactive Deep Colorization 的 PyTorch 重实现 richzhang.github.io/ideepcolor。
- beauty-net:一个简单、灵活且可扩展的 PyTorch 模板。它非常美观。
- OpenChem:面向计算化学和药物设计研究的深度学习工具包 mariewelt.github.io/OpenChem。
- torchani:基于 PyTorch 的精确神经网络势能 aiqm.github.io/torchani。
- PyTorch-LBFGS:L-BFGS 的 PyTorch 实现。
- gpytorch:一个高效且模块化的 PyTorch 高斯过程实现。
- hessian:PyTorch 中的 Hessian 矩阵计算。
- vel:深度学习研究中的速度指标。
- nonechucks:跳过 PyTorch DataLoader 中的不良样本,将变换用作过滤器等!
- torchstat:PyTorch 模型分析工具。
- QNNPACK:量化神经网络软件包——针对移动端优化的量化神经网络运算实现。
- torchdiffeq:具有完全 GPU 支持和 O(1) 内存反向传播的可微分常微分方程求解器。
- redner:一个可微分的蒙特卡洛路径追踪器。
- pixyz:一个以更简洁、直观和可扩展的方式开发深度生成模型的库。
- euclidesdb:一个多模型机器学习特征嵌入数据库 http://euclidesdb.readthedocs.io。
- pytorch2keras:将 PyTorch 动态图转换为 Keras 模型。
- salad:半监督学习和领域适应。
- netharn:用于 PyTorch 的参数化拟合和预测工具箱。
- dgl:一个基于现有深度学习框架构建的 Python 包,旨在简化图上的深度学习工作。http://dgl.ai。
- gandissect:基于 PyTorch 的工具,用于可视化和理解 GAN 的神经元结构。gandissect.csail.mit.edu。
- delira:一个轻量级框架,用于医学影像领域的快速原型化和训练深度神经网络 delira.rtfd.io。
- mushroom:一个用于强化学习实验的 Python 库。
- Xlearn:迁移学习库。
- geoopt:结合 PyTorch 优化器的黎曼流形自适应优化方法。
- vegans:一个提供多种现有 GAN 模型的 PyTorch 库。
- torchgeometry:TGM——PyTorch 几何学。
- AdverTorch:对抗鲁棒性(攻击/防御/训练)研究的工具箱。
- AdaBound:一种优化器,训练速度如 Adam,效果如 SGD。
- fenchel-young-losses:使用 Fenchel-Young 损失函数在 PyTorch/TensorFlow/scikit-learn 中进行概率分类。
- pytorch-OpCounter:统计你的 PyTorch 模型的浮点运算次数。
- Tor10:一个通用张量网络库,专为量子模拟设计,基于 PyTorch 构建。
- Catalyst:PyTorch DL & RL 研究的高级工具集。专注于可重复性、快速实验以及代码/想法的复用,使研究人员能够专注于创新而非重复编写训练循环。
- Ax:自适应实验平台。
- pywick:一个包含所有必要组件的 PyTorch 高级神经网络训练库。
- torchgpipe:PyTorch 中的 GPipe 实现 torchgpipe.readthedocs.io。
- hub:PyTorch Hub 是一个预训练模型仓库,旨在促进研究的可重复性。
- pytorch-lightning:PyTorch 的快速研究框架,相当于研究人员版本的 Keras。
- Tor10:一个通用张量网络库,专为量子模拟设计,基于 PyTorch 构建。
- tensorwatch:来自微软研究院的深度学习和强化学习调试、监控与可视化工具。
- wavetorch:数值求解波动方程并进行反向传播 arxiv.org/abs/1904.12831。
- diffdist:一个用于 PyTorch 的 Python 库,扩展了默认的
torch.autograd功能,并增加了进程间可微分通信的支持。 - torchprof:一个极低依赖的 PyTorch 模型逐层剖析库。
- osqpth:PyTorch 中的可微分 OSQP 求解器层。
- mctorch:一个用于深度学习的流形优化库。
- pytorch-hessian-eigenthings:利用 Hessian-向量乘积和随机幂迭代法高效分解 PyTorch Hessian 矩阵的特征值和特征向量。
- MinkowskiEngine:Minkowski Engine 是一个自动微分库,专为广义稀疏卷积和高维稀疏张量设计。
- pytorch-cpp-rl:PyTorch C++ 强化学习。
- pytorch-toolbelt:用于快速研发原型和 Kaggle 比赛的 PyTorch 扩展工具包。
- argus-tensor-stream:一个用于实时视频流解码并存储到 CUDA 内存的库 tensorstream.argus-ai.com。
- macarico:在 PyTorch 中学习搜索。
- rlpyt:PyTorch 中的强化学习。
- pywarm:一种更整洁的方式构建 PyTorch 神经网络。blue-season.github.io/pywarm。
- learn2learn:面向研究人员的 PyTorch 元学习框架 http://learn2learn.net。
- torchbeast:一个用于分布式强化学习的 PyTorch 平台。
- higher:一个允许用户获取跨越整个训练周期而非单个步骤的损失的更高阶梯度的 PyTorch 库。
- Torchelie:Torchelie 是一套用于 PyTorch 的实用函数、层、损失、模型、训练器等工具。torchelie.readthedocs.org。
- CrypTen:一个基于 PyTorch 的隐私保护机器学习框架,允许研究人员和开发者使用加密数据训练模型。CrypTen 目前支持安全多方计算作为其加密机制。
- cvxpylayers:一个用于在 PyTorch 中构建可微分凸优化层的 Python 库。
- RepDistiller:对比表征蒸馏 (CRD),以及近期知识蒸馏方法的基准测试。
- kaolin:一个旨在加速 3D 深度学习研究的 PyTorch 库。
- PySNN:高效的脉冲神经网络框架,基于 PyTorch 构建以利用 GPU 加速。
- sparktorch:在 Apache Spark 上训练和运行 PyTorch 模型。
- pytorch-metric-learning:在应用中使用度量学习的最简单方式。模块化、灵活且可扩展,基于 PyTorch 实现。
- autonomous-learning-library:一个用于构建深度强化学习智能体的 PyTorch 库。
- flambe:一个 ML 框架,用于加速研究及其走向生产的过程。flambe.ai。
- pytorch-optimizer:一系列现代优化算法的集合,包括:AccSGD、AdaBound、AdaMod、DiffGrad、Lamb、RAdam、Yogi。
- PyTorch-VAE:一组基于 PyTorch 的变分自编码器 (VAE)。
- ray:一个快速简单的框架,用于构建和运行分布式应用程序。Ray 自带 RLlib 可扩展强化学习库和 Tune 可扩展超参数调优库。ray.io。
- Pytorch Geometric Temporal:PyTorch Geometric 的时间序列扩展库。
- Poutyne:一个类似 Keras 的 PyTorch 框架,可以处理训练神经网络所需的大量样板代码。
- Pytorch-Toolbox:一个面向 PyTorch 的工具箱项目,旨在让编写 PyTorch 代码更加简单、易读和简洁。
- Pytorch-contrib:包含了对近期机器学习论文中提出的想法的评审实现。
- EfficientNet PyTorch:包含逐层对应的 EfficientNet 的 PyTorch 重实现,以及预训练模型和示例。
- PyTorch/XLA:一个使用 XLA 深度学习编译器将 PyTorch 深度学习框架与 Cloud TPU 连接的 Python 包。
- webdataset:WebDataset 是一个 PyTorch 数据集(IterableDataset)实现,提供对存储在 POSIX tar 归档中的数据集的高效访问。
- volksdep:一个开源工具箱,用于部署和加速 PyTorch、Onnx 和 Tensorflow 模型,并结合 TensorRT 使用。
- PyTorch-StudioGAN:StudioGAN 是一个 PyTorch 库,提供了用于条件/无条件图像生成的代表性生成对抗网络 (GAN) 实现。StudioGAN 旨在为现代 GAN 提供一个统一的比较和分析平台,方便机器学习研究人员评估新想法。
- torchdrift:漂移检测库。
- accelerate:一种简单的方法,用于使用多 GPU、TPU 或混合精度训练和使用 PyTorch 模型。
- lightning-transformers:一个灵活的接口,用于借助 PyTorch Lightning、Transformers 和 Hydra 进行高性能 SOTA 变换器研究。
- Flower:一种统一的联邦学习、分析和评估方法,允许联邦任何机器学习任务。
- lightning-flash:Flash 是一系列用于快速原型化、基线建立和可扩展深度学习模型微调的任务集合,基于 PyTorch Lightning 构建。
- Pytorch Geometric Signed Directed:PyTorch Geometric 的有符号和有向扩展库。
- Koila:一个简单的 PyTorch 包装器,用于防止 CUDA 内存不足问题。
- Renate:一个用于真实世界持续学习的库。
- ANEE:面向 PyTorch 转换器的自适应神经执行引擎。提供按令牌动态跳过层、基于性能分析的门控机制以及 KV 缓存安全的稀疏推理等功能。
教程、书籍和示例
- 实用 PyTorch:讲解不同 RNN 模型的教程
- DeepLearningForNLPInPytorch:一个关于深度学习的 IPython Notebook 教程,重点在于自然语言处理。
- pytorch-tutorial:面向研究人员的 PyTorch 深度学习教程。
- pytorch-exercises:PyTorch 练习集。
- pytorch tutorials:各种 PyTorch 教程。
- pytorch examples:展示 PyTorch 使用示例的仓库。
- pytorch practice:一些 PyTorch 示例脚本。
- pytorch mini tutorials:基于 Alec Radford 的 Theano 教程改编的 PyTorch 极简教程。
- pytorch text classification:在 PyTorch 中实现的一个基于 CNN 的简单文本分类模型。
- cats vs dogs:用于 Kaggle 竞赛“Dogs vs. Cats Redux: Kernels Edition”的 PyTorch 网络微调示例。目前在排行榜上排名第 27 位(0.05074)。
- convnet:这是一个针对多种数据集(ImageNet、Cifar10、Cifar100、MNIST)的深度卷积神经网络完整训练示例。
- pytorch-generative-adversarial-networks:使用 PyTorch 实现的简单生成对抗网络(GAN)。
- pytorch containers:该仓库旨在通过提供 Torch Table Layers 的 PyTorch 实现列表,帮助原 Torch 用户更顺利地过渡到无容器的 PyTorch 世界。
- T-SNE in pytorch:PyTorch 中的 t-SNE 实验。
- AAE_pytorch:使用 PyTorch 实现的对抗自编码器。
- Kind_PyTorch_Tutorial:面向初学者的友好 PyTorch 教程。
- pytorch-poetry-gen:基于 PyTorch 的字符级 RNN。
- pytorch-REINFORCE:PyTorch 版 REINFORCE 算法实现。该仓库支持 OpenAI Gym 中的连续和离散环境。
- PyTorch-Tutorial:轻松快速构建你的神经网络 https://morvanzhou.github.io/tutorials/
- pytorch-intro:几段脚本,用于演示如何在 PyTorch 中进行 CNN 和 RNN 操作。
- pytorch-classification:一个统一的框架,用于 CIFAR-10/100 和 ImageNet 上的图像分类任务。
- pytorch_notebooks - hardmaru:用 NumPy 和 PyTorch 编写的随机教程。
- pytorch_tutoria-quick:快速 PyTorch 入门与教程。目标用户是希望尝试新框架的计算机视觉、图形学和机器学习研究人员。
- Pytorch_fine_tuning_Tutorial:关于在 PyTorch 中进行微调或迁移学习的简短教程。
- pytorch_exercises:PyTorch 练习。
- traffic-sign-detection:nyu-cv-fall-2017 示例。
- mss_pytorch:基于循环推理和跳跃滤波连接的歌声分离——PyTorch 实现。演示地址:js-mim.github.io/mss_pytorch。
- DeepNLP-models-Pytorch:斯坦福大学 CS-224n 课程中各种深度 NLP 模型的 PyTorch 实现。
- Mila introductory tutorials:为 MILA 新生提供的各类入门教程。
- pytorch.rl.learning:用于使用 PyTorch 学习强化学习。
- minimal-seq2seq:在 PyTorch 中实现的带有注意力机制的极简序列到序列模型,用于神经机器翻译。
- tensorly-notebooks:使用 TensorLy 库在 Python 中进行张量方法操作 tensorly.github.io/dev。
- pytorch_bits:与时间序列预测相关的示例。
- skip-thoughts:在 PyTorch 中实现的 Skip-Thought 向量。
- video-caption-pytorch:用于视频字幕生成的 PyTorch 代码。
- Capsule-Network-Tutorial:易于理解的 PyTorch 胶囊网络教程。
- code-of-learn-deep-learning-with-pytorch:这是书籍《用 PyTorch 学习深度学习》的配套代码 item.jd.com/17915495606.html。
- RL-Adventure:易于跟随的 PyTorch 步骤式深度 Q 学习教程,代码清晰易读。
- accelerated_dl_pytorch:在亚特兰大 Jupyter Day II 上加速的 PyTorch 深度学习实践。
- RL-Adventure-2:PyTorch 教程,内容包括:演员-评论家算法 / 近端策略优化 / ACER / DDPG / 双重决斗 DDPG / 软演员-评论家算法 / 生成对抗模仿学习 / 回顾性经验回放。
- 50 行代码实现的生成对抗网络(GAN)(PyTorch)
- 使用 PyTorch 的对抗自编码器
- 使用 PyTorch 进行迁移学习
- 如何在 PyTorch 中实现 YOLO 目标检测器
- 推荐系统中的 PyTorch 入门
- 面向 NumPy 用户的 PyTorch
- PyTorch Tutorial:中文版 PyTorch 教程。
- grokking-pytorch:PyTorch 指南。
- PyTorch-Deep-Learning-Minicourse:PyTorch 深度学习迷你课程。
- pytorch-custom-dataset-examples:一些用于 PyTorch 的自定义数据集示例。
- 用于基于序列的推荐系统的乘法 LSTM
- deeplearning.ai-pytorch:Coursera 的深度学习(deeplearning.ai)专项课程的 PyTorch 实现。
- MNIST_Pytorch_python_and_capi:这是一个示例,说明如何用 Python 训练 MNIST 网络,并使用 PyTorch 1.0 在 C++ 中运行。
- torch_light:教程和示例涵盖强化学习、NLP 和 CV。
- portrain-gan:用于解码(并几乎编码)艺术 DC-GAN 的 Portrait GAN 隐空间的 PyTorch 代码。
- mri-analysis-pytorch:使用 PyTorch 和 MedicalTorch 进行 MRI 分析。
- cifar10-fast: 展示了如何按照这篇博客系列[https://www.myrtle.ai/2018/09/24/how_to_train_your_resnet/]的描述,在 CIFAR10 数据集上训练一个小 ResNet,使其在 79 秒内达到 94% 的测试准确率。
- PyTorch 深度学习入门:由 Udacity 和 Facebook 联合提供的免费课程,包含良好的 PyTorch 入门介绍,并采访了 PyTorch 的原始作者之一 Soumith Chintala。
- pytorch-sentiment-analysis:关于如何开始使用 PyTorch 和 TorchText 进行情感分析的教程。
- pytorch-image-models:PyTorch 图像模型、脚本及预训练权重——(SE)ResNet/ResNeXT、DPN、EfficientNet、MobileNet-V3/V2/V1、MNASNet、Single-Path NAS、FBNet 等。
- CIFAR-ZOO:多个 CNN 架构及改进方法的 PyTorch 实现,取得了最先进的结果。
- d2l-pytorch:试图将伯克利 STAT 157(2019 年春季)教材《深入深度学习》中的代码修改为 PyTorch 格式。
- thinking-in-tensors-writing-in-pytorch:以张量思维、PyTorch 编写(一本动手实践的深度学习入门)。
- NER-BERT-pytorch:使用 Google AI 预训练的 BERT 模型解决命名实体识别任务的 PyTorch 解法。
- pytorch-sync-batchnorm-example:如何在 PyTorch 中使用跨副本/同步批归一化。
- SentimentAnalysis:基于斯坦福情感树库对 BERT 进行微调训练的情感分析神经网络,感谢Hugging Face的 Transformers 库。
- pytorch-cpp:面向深度学习研究人员的 PyTorch 教程 C++ 实现(基于yunjey/pytorch-tutorial中的 Python 教程)。
- 深度学习与 PyTorch:从零到 GAN:一系列互动且注重编码的教程,介绍深度学习与 PyTorch(视频)。
- 深度学习与 PyTorch:本书教你如何使用 Python 和 PyTorch 实现深度学习算法,书中包含一个案例研究:构建一种能够利用 CT 扫描检测恶性肺肿瘤的算法。
- 使用 PyTorch 和 AWS 实现无服务器机器学习:这是一本指南,教你如何利用 AWS、Azure 或 GCP 等主要云服务提供商的无服务器功能,将实验性的 PyTorch 机器学习代码投入生产。
- LabML NN:一组神经网络架构和算法的 PyTorch 实现,并附有并排注释。
- 使用 Flower 将你的 PyTorch 示例联邦化:此示例展示了如何将现有的集中式 PyTorch 机器学习项目通过 Flower 实现联邦化。使用了 Cifar-10 数据集和卷积神经网络(CNN)。
- 人工智能背后的数学:一本免费的 FreeCodeCamp 书籍,从工程角度用通俗易懂的语言讲解 AI 背后的数学知识。内容涵盖线性代数、微积分、概率统计以及优化理论,并配有类比、实际应用和 Python 代码示例。
论文实现
- google_evolution: This implements one of result networks from Large-scale evolution of image classifiers by Esteban Real, et. al.
- pyscatwave: Fast Scattering Transform with CuPy/PyTorch,read the paper here
- scalingscattering: Scaling The Scattering Transform : Deep Hybrid Networks.
- deep-auto-punctuation: a pytorch implementation of auto-punctuation learned character by character.
- Realtime_Multi-Person_Pose_Estimation: This is a pytorch version of Realtime_Multi-Person_Pose_Estimation, origin code is here .
- PyTorch-value-iteration-networks: PyTorch implementation of the Value Iteration Networks (NIPS '16) paper
- pytorch_Highway: Highway network implemented in pytorch.
- pytorch_NEG_loss: NEG loss implemented in pytorch.
- pytorch_RVAE: Recurrent Variational Autoencoder that generates sequential data implemented in pytorch.
- pytorch_TDNN: Time Delayed NN implemented in pytorch.
- eve.pytorch: An implementation of Eve Optimizer, proposed in Imploving Stochastic Gradient Descent with Feedback, Koushik and Hayashi, 2016.
- e2e-model-learning: Task-based end-to-end model learning.
- pix2pix-pytorch: PyTorch implementation of "Image-to-Image Translation Using Conditional Adversarial Networks".
- Single Shot MultiBox Detector: A PyTorch Implementation of Single Shot MultiBox Detector.
- DiscoGAN: PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"
- official DiscoGAN implementation: Official implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks".
- pytorch-es: This is a PyTorch implementation of Evolution Strategies .
- piwise: Pixel-wise segmentation on VOC2012 dataset using pytorch.
- pytorch-dqn: Deep Q-Learning Network in pytorch.
- neuraltalk2-pytorch: image captioning model in pytorch(finetunable cnn in branch with_finetune)
- vnet.pytorch: A Pytorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation.
- pytorch-fcn: PyTorch implementation of Fully Convolutional Networks.
- WideResNets: WideResNets for CIFAR10/100 implemented in PyTorch. This implementation requires less GPU memory than what is required by the official Torch implementation: https://github.com/szagoruyko/wide-residual-networks .
- pytorch_highway_networks: Highway networks implemented in PyTorch.
- pytorch-NeuCom: Pytorch implementation of DeepMind's differentiable neural computer paper.
- captionGen: Generate captions for an image using PyTorch.
- AnimeGAN: A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.
- Cnn-text classification: This is the implementation of Kim's Convolutional Neural Networks for Sentence Classification paper in PyTorch.
- deepspeech2: Implementation of DeepSpeech2 using Baidu Warp-CTC. Creates a network based on the DeepSpeech2 architecture, trained with the CTC activation function.
- seq2seq: This repository contains implementations of Sequence to Sequence (Seq2Seq) models in PyTorch
- Asynchronous Advantage Actor-Critic in PyTorch: This is PyTorch implementation of A3C as described in Asynchronous Methods for Deep Reinforcement Learning. Since PyTorch has a easy method to control shared memory within multiprocess, we can easily implement asynchronous method like A3C.
- densenet: This is a PyTorch implementation of the DenseNet-BC architecture as described in the paper Densely Connected Convolutional Networks by G. Huang, Z. Liu, K. Weinberger, and L. van der Maaten. This implementation gets a CIFAR-10+ error rate of 4.77 with a 100-layer DenseNet-BC with a growth rate of 12. Their official implementation and links to many other third-party implementations are available in the liuzhuang13/DenseNet repo on GitHub.
- nninit: Weight initialization schemes for PyTorch nn.Modules. This is a port of the popular nninit for Torch7 by @kaixhin.
- faster rcnn: This is a PyTorch implementation of Faster RCNN. This project is mainly based on py-faster-rcnn and TFFRCNN.For details about R-CNN please refer to the paper Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun.
- doomnet: PyTorch's version of Doom-net implementing some RL models in ViZDoom environment.
- flownet: Pytorch implementation of FlowNet by Dosovitskiy et al.
- sqeezenet: Implementation of Squeezenet in pytorch, #### pretrained models on CIFAR10 data to come Plan to train the model on cifar 10 and add block connections too.
- WassersteinGAN: wassersteinGAN in pytorch.
- optnet: This repository is by Brandon Amos and J. Zico Kolter and contains the PyTorch source code to reproduce the experiments in our paper OptNet: Differentiable Optimization as a Layer in Neural Networks.
- qp solver: A fast and differentiable QP solver for PyTorch. Crafted by Brandon Amos and J. Zico Kolter.
- Continuous Deep Q-Learning with Model-based Acceleration : Reimplementation of Continuous Deep Q-Learning with Model-based Acceleration.
- Learning to learn by gradient descent by gradient descent: PyTorch implementation of Learning to learn by gradient descent by gradient descent.
- fast-neural-style: pytorch implementation of fast-neural-style, The model uses the method described in Perceptual Losses for Real-Time Style Transfer and Super-Resolution along with Instance Normalization.
- PytorchNeuralStyleTransfer: Implementation of Neural Style Transfer in Pytorch.
- Fast Neural Style for Image Style Transform by Pytorch: Fast Neural Style for Image Style Transform by Pytorch .
- neural style transfer: An introduction to PyTorch through the Neural-Style algorithm (https://arxiv.org/abs/1508.06576) developed by Leon A. Gatys, Alexander S. Ecker and Matthias Bethge.
- VIN_PyTorch_Visdom: PyTorch implementation of Value Iteration Networks (VIN): Clean, Simple and Modular. Visualization in Visdom.
- YOLO2: YOLOv2 in PyTorch.
- attention-transfer: Attention transfer in pytorch, read the paper here.
- SVHNClassifier: A PyTorch implementation of Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networks.
- pytorch-deform-conv: PyTorch implementation of Deformable Convolution.
- BEGAN-pytorch: PyTorch implementation of BEGAN: Boundary Equilibrium Generative Adversarial Networks.
- treelstm.pytorch: Tree LSTM implementation in PyTorch.
- AGE: Code for paper "Adversarial Generator-Encoder Networks" by Dmitry Ulyanov, Andrea Vedaldi and Victor Lempitsky which can be found here
- ResNeXt.pytorch: Reproduces ResNet-V3 (Aggregated Residual Transformations for Deep Neural Networks) with pytorch.
- pytorch-rl: Deep Reinforcement Learning with pytorch & visdom
- Deep-Leafsnap: LeafSnap replicated using deep neural networks to test accuracy compared to traditional computer vision methods.
- pytorch-CycleGAN-and-pix2pix: PyTorch implementation for both unpaired and paired image-to-image translation.
- A3C-PyTorch:PyTorch implementation of Advantage async actor-critic Algorithms (A3C) in PyTorch
- pytorch-value-iteration-networks: Pytorch implementation of Value Iteration Networks (NIPS 2016 best paper)
- PyTorch-Style-Transfer: PyTorch Implementation of Multi-style Generative Network for Real-time Transfer
- pytorch-deeplab-resnet: pytorch-deeplab-resnet-model.
- pointnet.pytorch: pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593
- pytorch-playground: Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet).
- pytorch-dnc: Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom.
- pytorch_image_classifier: Minimal But Practical Image Classifier Pipline Using Pytorch, Finetune on ResNet18, Got 99% Accuracy on Own Small Datasets.
- mnist-svhn-transfer: PyTorch Implementation of CycleGAN and SGAN for Domain Transfer (Minimal).
- pytorch-yolo2: pytorch-yolo2
- dni: Implement Decoupled Neural Interfaces using Synthetic Gradients in Pytorch
- wgan-gp: A pytorch implementation of Paper "Improved Training of Wasserstein GANs".
- pytorch-seq2seq-intent-parsing: Intent parsing and slot filling in PyTorch with seq2seq + attention
- pyTorch_NCE: An implementation of the Noise Contrastive Estimation algorithm for pyTorch. Working, yet not very efficient.
- molencoder: Molecular AutoEncoder in PyTorch
- GAN-weight-norm: Code for "On the Effects of Batch and Weight Normalization in Generative Adversarial Networks"
- lgamma: Implementations of polygamma, lgamma, and beta functions for PyTorch
- bigBatch: Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training of neural networks"
- rl_a3c_pytorch: Reinforcement learning with implementation of A3C LSTM for Atari 2600.
- pytorch-retraining: Transfer Learning Shootout for PyTorch's model zoo (torchvision)
- nmp_qc: Neural Message Passing for Computer Vision
- grad-cam: Pytorch implementation of Grad-CAM
- pytorch-trpo: PyTorch Implementation of Trust Region Policy Optimization (TRPO)
- pytorch-explain-black-box: PyTorch implementation of Interpretable Explanations of Black Boxes by Meaningful Perturbation
- vae_vpflows: Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling https://jmtomczak.github.io/deebmed.html
- relational-networks: Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks) https://arxiv.org/pdf/1706.01427.pdf
- vqa.pytorch: Visual Question Answering in Pytorch
- end-to-end-negotiator: Deal or No Deal? End-to-End Learning for Negotiation Dialogues
- odin-pytorch: Principled Detection of Out-of-Distribution Examples in Neural Networks.
- FreezeOut: Accelerate Neural Net Training by Progressively Freezing Layers.
- ARAE: Code for the paper "Adversarially Regularized Autoencoders for Generating Discrete Structures" by Zhao, Kim, Zhang, Rush and LeCun.
- forward-thinking-pytorch: Pytorch implementation of "Forward Thinking: Building and Training Neural Networks One Layer at a Time" https://arxiv.org/pdf/1706.02480.pdf
- context_encoder_pytorch: PyTorch Implement of Context Encoders
- attention-is-all-you-need-pytorch: A PyTorch implementation of the Transformer model in "Attention is All You Need".https://github.com/thnkim/OpenFacePytorch
- OpenFacePytorch: PyTorch module to use OpenFace's nn4.small2.v1.t7 model
- neural-combinatorial-rl-pytorch: PyTorch implementation of Neural Combinatorial Optimization with Reinforcement Learning.
- pytorch-nec: PyTorch Implementation of Neural Episodic Control (NEC)
- seq2seq.pytorch: Sequence-to-Sequence learning using PyTorch
- Pytorch-Sketch-RNN: a pytorch implementation of arxiv.org/abs/1704.03477
- pytorch-pruning: PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
- DrQA: A pytorch implementation of Reading Wikipedia to Answer Open-Domain Questions.
- YellowFin_Pytorch: auto-tuning momentum SGD optimizer
- samplernn-pytorch: PyTorch implementation of SampleRNN: An Unconditional End-to-End Neural Audio Generation Model.
- AEGeAN: Deeper DCGAN with AE stabilization
- /pytorch-SRResNet: pytorch implementation for Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network arXiv:1609.04802v2
- vsepp: Code for the paper "VSE++: Improved Visual Semantic Embeddings"
- Pytorch-DPPO: Pytorch implementation of Distributed Proximal Policy Optimization: arxiv.org/abs/1707.02286
- UNIT: PyTorch Implementation of our Coupled VAE-GAN algorithm for Unsupervised Image-to-Image Translation
- efficient_densenet_pytorch: A memory-efficient implementation of DenseNets
- tsn-pytorch: Temporal Segment Networks (TSN) in PyTorch.
- SMASH: An experimental technique for efficiently exploring neural architectures.
- pytorch-retinanet: RetinaNet in PyTorch
- biogans: Implementation supporting the ICCV 2017 paper "GANs for Biological Image Synthesis".
- Semantic Image Synthesis via Adversarial Learning: A PyTorch implementation of the paper "Semantic Image Synthesis via Adversarial Learning" in ICCV 2017.
- fmpytorch: A PyTorch implementation of a Factorization Machine module in cython.
- ORN: A PyTorch implementation of the paper "Oriented Response Networks" in CVPR 2017.
- pytorch-maml: PyTorch implementation of MAML: arxiv.org/abs/1703.03400
- pytorch-generative-model-collections: Collection of generative models in Pytorch version.
- vqa-winner-cvprw-2017: Pytorch Implementation of winner from VQA Chllange Workshop in CVPR'17.
- tacotron_pytorch: PyTorch implementation of Tacotron speech synthesis model.
- pspnet-pytorch: PyTorch implementation of PSPNet segmentation network
- LM-LSTM-CRF: Empower Sequence Labeling with Task-Aware Language Model http://arxiv.org/abs/1709.04109
- face-alignment: Pytorch implementation of the paper "How far are we from solving the 2D & 3D Face Alignment problem? (and a dataset of 230,000 3D facial landmarks)", ICCV 2017
- DepthNet: PyTorch DepthNet Training on Still Box dataset.
- EDSR-PyTorch: PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
- e2c-pytorch: Embed to Control implementation in PyTorch.
- 3D-ResNets-PyTorch: 3D ResNets for Action Recognition.
- bandit-nmt: This is code repo for our EMNLP 2017 paper "Reinforcement Learning for Bandit Neural Machine Translation with Simulated Human Feedback", which implements the A2C algorithm on top of a neural encoder-decoder model and benchmarks the combination under simulated noisy rewards.
- pytorch-a2c-ppo-acktr: PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO) and Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR).
- zalando-pytorch: Various experiments on the Fashion-MNIST dataset from Zalando.
- sphereface_pytorch: A PyTorch Implementation of SphereFace.
- Categorical DQN: A PyTorch Implementation of Categorical DQN from A Distributional Perspective on Reinforcement Learning.
- pytorch-ntm: pytorch ntm implementation.
- mask_rcnn_pytorch: Mask RCNN in PyTorch.
- graph_convnets_pytorch: PyTorch implementation of graph ConvNets, NIPS’16
- pytorch-faster-rcnn: A pytorch implementation of faster RCNN detection framework based on Xinlei Chen's tf-faster-rcnn.
- torchMoji: A pyTorch implementation of the DeepMoji model: state-of-the-art deep learning model for analyzing sentiment, emotion, sarcasm etc.
- semantic-segmentation-pytorch: Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
- pytorch-qrnn: PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM
- pytorch-sgns: Skipgram Negative Sampling in PyTorch.
- SfmLearner-Pytorch : Pytorch version of SfmLearner from Tinghui Zhou et al.
- deformable-convolution-pytorch: PyTorch implementation of Deformable Convolution.
- skip-gram-pytorch: A complete pytorch implementation of skipgram model (with subsampling and negative sampling). The embedding result is tested with Spearman's rank correlation.
- stackGAN-v2: Pytorch implementation for reproducing StackGAN_v2 results in the paper StackGAN++: Realistic Image Synthesis with Stacked Generative Adversarial Networks by Han Zhang*, Tao Xu*, Hongsheng Li, Shaoting Zhang, Xiaogang Wang, Xiaolei Huang, Dimitris Metaxas.
- self-critical.pytorch: Unofficial pytorch implementation for Self-critical Sequence Training for Image Captioning.
- pygcn: Graph Convolutional Networks in PyTorch.
- dnc: Differentiable Neural Computers, for Pytorch
- prog_gans_pytorch_inference: PyTorch inference for "Progressive Growing of GANs" with CelebA snapshot.
- pytorch-capsule: Pytorch implementation of Hinton's Dynamic Routing Between Capsules.
- PyramidNet-PyTorch: A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, arxiv.org/abs/1610.02915)
- radio-transformer-networks: A PyTorch implementation of Radio Transformer Networks from the paper "An Introduction to Deep Learning for the Physical Layer". arxiv.org/abs/1702.00832
- honk: PyTorch reimplementation of Google's TensorFlow CNNs for keyword spotting.
- DeepCORAL: A PyTorch implementation of 'Deep CORAL: Correlation Alignment for Deep Domain Adaptation.', ECCV 2016
- pytorch-pose: A PyTorch toolkit for 2D Human Pose Estimation.
- lang-emerge-parlai: Implementation of EMNLP 2017 Paper "Natural Language Does Not Emerge 'Naturally' in Multi-Agent Dialog" using PyTorch and ParlAI
- Rainbow: Rainbow: Combining Improvements in Deep Reinforcement Learning
- pytorch_compact_bilinear_pooling v1: This repository has a pure Python implementation of Compact Bilinear Pooling and Count Sketch for PyTorch.
- CompactBilinearPooling-Pytorch v2: (Yang Gao, et al.) A Pytorch Implementation for Compact Bilinear Pooling.
- FewShotLearning: Pytorch implementation of the paper "Optimization as a Model for Few-Shot Learning"
- meProp: Codes for "meProp: Sparsified Back Propagation for Accelerated Deep Learning with Reduced Overfitting".
- SFD_pytorch: A PyTorch Implementation of Single Shot Scale-invariant Face Detector.
- GradientEpisodicMemory: Continuum Learning with GEM: Gradient Episodic Memory. https://arxiv.org/abs/1706.08840
- DeblurGAN: Pytorch implementation of the paper DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networks.
- StarGAN: StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Tranlsation.
- CapsNet-pytorch: PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules.
- CondenseNet: CondenseNet: An Efficient DenseNet using Learned Group Convolutions.
- deep-image-prior: Image restoration with neural networks but without learning.
- deep-head-pose: Deep Learning Head Pose Estimation using PyTorch.
- Random-Erasing: This code has the source code for the paper "Random Erasing Data Augmentation".
- FaderNetworks: Fader Networks: Manipulating Images by Sliding Attributes - NIPS 2017
- FlowNet 2.0: FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
- pix2pixHD: Synthesizing and manipulating 2048x1024 images with conditional GANs tcwang0509.github.io/pix2pixHD
- pytorch-smoothgrad: SmoothGrad implementation in PyTorch
- RetinaNet: An implementation of RetinaNet in PyTorch.
- faster-rcnn.pytorch: This project is a faster faster R-CNN implementation, aimed to accelerating the training of faster R-CNN object detection models.
- mixup_pytorch: A PyTorch implementation of the paper Mixup: Beyond Empirical Risk Minimization in PyTorch.
- inplace_abn: In-Place Activated BatchNorm for Memory-Optimized Training of DNNs
- pytorch-pose-hg-3d: PyTorch implementation for 3D human pose estimation
- nmn-pytorch: Neural Module Network for VQA in Pytorch.
- bytenet: Pytorch implementation of bytenet from "Neural Machine Translation in Linear Time" paper
- bottom-up-attention-vqa: vqa, bottom-up-attention, pytorch
- yolo2-pytorch: The YOLOv2 is one of the most popular one-stage object detector. This project adopts PyTorch as the developing framework to increase productivity, and utilize ONNX to convert models into Caffe 2 to benifit engineering deployment.
- reseg-pytorch: PyTorch Implementation of ReSeg (arxiv.org/pdf/1511.07053.pdf)
- binary-stochastic-neurons: Binary Stochastic Neurons in PyTorch.
- pytorch-pose-estimation: PyTorch Implementation of Realtime Multi-Person Pose Estimation project.
- interaction_network_pytorch: Pytorch Implementation of Interaction Networks for Learning about Objects, Relations and Physics.
- NoisyNaturalGradient: Pytorch Implementation of paper "Noisy Natural Gradient as Variational Inference".
- ewc.pytorch: An implementation of Elastic Weight Consolidation (EWC), proposed in James Kirkpatrick et al. Overcoming catastrophic forgetting in neural networks 2016(10.1073/pnas.1611835114).
- pytorch-zssr: PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning
- deep_image_prior: An implementation of image reconstruction methods from Deep Image Prior (Ulyanov et al., 2017) in PyTorch.
- pytorch-transformer: pytorch implementation of Attention is all you need.
- DeepRL-Grounding: This is a PyTorch implementation of the AAAI-18 paper Gated-Attention Architectures for Task-Oriented Language Grounding
- deep-forecast-pytorch: Wind Speed Prediction using LSTMs in PyTorch (arxiv.org/pdf/1707.08110.pdf)
- cat-net: Canonical Appearance Transformations
- minimal_glo: Minimal PyTorch implementation of Generative Latent Optimization from the paper "Optimizing the Latent Space of Generative Networks"
- LearningToCompare-Pytorch: Pytorch Implementation for Paper: Learning to Compare: Relation Network for Few-Shot Learning.
- poincare-embeddings: PyTorch implementation of the NIPS-17 paper "Poincaré Embeddings for Learning Hierarchical Representations".
- pytorch-trpo(Hessian-vector product version): This is a PyTorch implementation of "Trust Region Policy Optimization (TRPO)" with exact Hessian-vector product instead of finite differences approximation.
- ggnn.pytorch: A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN).
- visual-interaction-networks-pytorch: This's an implementation of deepmind Visual Interaction Networks paper using pytorch
- adversarial-patch: PyTorch implementation of adversarial patch.
- Prototypical-Networks-for-Few-shot-Learning-PyTorch: Implementation of Prototypical Networks for Few Shot Learning (arxiv.org/abs/1703.05175) in Pytorch
- Visual-Feature-Attribution-Using-Wasserstein-GANs-Pytorch: Implementation of Visual Feature Attribution using Wasserstein GANs (arxiv.org/abs/1711.08998) in PyTorch.
- PhotographicImageSynthesiswithCascadedRefinementNetworks-Pytorch: Photographic Image Synthesis with Cascaded Refinement Networks - Pytorch Implementation
- ENAS-pytorch: PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing".
- Neural-IMage-Assessment: A PyTorch Implementation of Neural IMage Assessment.
- proxprop: Proximal Backpropagation - a neural network training algorithm that takes implicit instead of explicit gradient steps.
- FastPhotoStyle: A Closed-form Solution to Photorealistic Image Stylization
- Deep-Image-Analogy-PyTorch: A python implementation of Deep-Image-Analogy based on pytorch.
- Person-reID_pytorch: PyTorch for Person re-ID.
- pt-dilate-rnn: Dilated RNNs in pytorch.
- pytorch-i-revnet: Pytorch implementation of i-RevNets.
- OrthNet: TensorFlow and PyTorch layers for generating Orthogonal Polynomials.
- DRRN-pytorch: An implementation of Deep Recursive Residual Network for Super Resolution (DRRN), CVPR 2017
- shampoo.pytorch: An implementation of shampoo.
- Neural-IMage-Assessment 2: A PyTorch Implementation of Neural IMage Assessment.
- TCN: Sequence modeling benchmarks and temporal convolutional networks locuslab/TCN
- DCC: This repository contains the source code and data for reproducing results of Deep Continuous Clustering paper.
- packnet: Code for PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruning arxiv.org/abs/1711.05769
- PyTorch-progressive_growing_of_gans: PyTorch implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation.
- nonauto-nmt: PyTorch Implementation of "Non-Autoregressive Neural Machine Translation"
- PyTorch-GAN: PyTorch implementations of Generative Adversarial Networks.
- PyTorchWavelets: PyTorch implementation of the wavelet analysis found in Torrence and Compo (1998)
- pytorch-made: MADE (Masked Autoencoder Density Estimation) implementation in PyTorch
- VRNN: Pytorch implementation of the Variational RNN (VRNN), from A Recurrent Latent Variable Model for Sequential Data.
- flow: Pytorch implementation of ICLR 2018 paper Deep Learning for Physical Processes: Integrating Prior Scientific Knowledge.
- deepvoice3_pytorch: PyTorch implementation of convolutional networks-based text-to-speech synthesis models
- psmm: imlementation of the the Pointer Sentinel Mixture Model, as described in the paper by Stephen Merity et al.
- tacotron2: Tacotron 2 - PyTorch implementation with faster-than-realtime inference.
- AccSGD: Implements pytorch code for the Accelerated SGD algorithm.
- QANet-pytorch: an implementation of QANet with PyTorch (EM/F1 = 70.5/77.2 after 20 epoches for about 20 hours on one 1080Ti card.)
- ConvE: Convolutional 2D Knowledge Graph Embeddings
- Structured-Self-Attention: Implementation for the paper A Structured Self-Attentive Sentence Embedding, which is published in ICLR 2017: arxiv.org/abs/1703.03130 .
- graphsage-simple: Simple reference implementation of GraphSAGE.
- Detectron.pytorch: A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
- R2Plus1D-PyTorch: PyTorch implementation of the R2Plus1D convolution based ResNet architecture described in the paper "A Closer Look at Spatiotemporal Convolutions for Action Recognition"
- StackNN: A PyTorch implementation of differentiable stacks for use in neural networks.
- translagent: Code for Emergent Translation in Multi-Agent Communication.
- ban-vqa: Bilinear attention networks for visual question answering.
- pytorch-openai-transformer-lm: This is a PyTorch implementation of the TensorFlow code provided with OpenAI's paper "Improving Language Understanding by Generative Pre-Training" by Alec Radford, Karthik Narasimhan, Tim Salimans and Ilya Sutskever.
- T2F: Text-to-Face generation using Deep Learning. This project combines two of the recent architectures StackGAN and ProGAN for synthesizing faces from textual descriptions.
- pytorch - fid: A Port of Fréchet Inception Distance (FID score) to PyTorch
- vae_vpflows:Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling jmtomczak.github.io/deebmed.html
- CoordConv-pytorch: Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' paper. (arxiv.org/pdf/1807.03247.pdf)
- SDPoint: Implementation of "Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks", published in CVPR 2018.
- SRDenseNet-pytorch: SRDenseNet-pytorch(ICCV_2017)
- GAN_stability: Code for paper "Which Training Methods for GANs do actually Converge? (ICML 2018)"
- Mask-RCNN: A PyTorch implementation of the architecture of Mask RCNN, serves as an introduction to working with PyTorch
- pytorch-coviar: Compressed Video Action Recognition
- PNASNet.pytorch: PyTorch implementation of PNASNet-5 on ImageNet.
- NALU-pytorch: Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units arxiv.org/pdf/1808.00508.pdf
- LOLA_DiCE: Pytorch implementation of LOLA (arxiv.org/abs/1709.04326) using DiCE (arxiv.org/abs/1802.05098)
- generative-query-network-pytorch: Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"
- pytorch_hmax: Implementation of the HMAX model of vision in PyTorch.
- FCN-pytorch-easiest: trying to be the most easiest and just get-to-use pytorch implementation of FCN (Fully Convolotional Networks)
- transducer: A Fast Sequence Transducer Implementation with PyTorch Bindings.
- AVO-pytorch: Implementation of Adversarial Variational Optimization in PyTorch.
- HCN-pytorch: A pytorch reimplementation of { Co-occurrence Feature Learning from Skeleton Data for Action Recognition and Detection with Hierarchical Aggregation }.
- binary-wide-resnet: PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnel (ICLR 2018)
- piggyback: Code for Piggyback: Adapting a Single Network to Multiple Tasks by Learning to Mask Weights arxiv.org/abs/1801.06519
- vid2vid: Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
- poisson-convolution-sum: Implements an infinite sum of poisson-weighted convolutions
- tbd-nets: PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning" arxiv.org/abs/1803.05268
- attn2d: Pervasive Attention: 2D Convolutional Networks for Sequence-to-Sequence Prediction
- yolov3: YOLOv3: Training and inference in PyTorch pjreddie.com/darknet/yolo
- deep-dream-in-pytorch: Pytorch implementation of the DeepDream computer vision algorithm.
- pytorch-flows: PyTorch implementations of algorithms for density estimation
- quantile-regression-dqn-pytorch: Quantile Regression DQN a Minimal Working Example
- relational-rnn-pytorch: An implementation of DeepMind's Relational Recurrent Neural Networks in PyTorch.
- DEXTR-PyTorch: Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr
- PyTorch_GBW_LM: PyTorch Language Model for Google Billion Word Dataset.
- Pytorch-NCE: The Noise Contrastive Estimation for softmax output written in Pytorch
- generative-models: Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN.
- convnet-aig: PyTorch implementation for Convolutional Networks with Adaptive Inference Graphs.
- integrated-gradient-pytorch: This is the pytorch implementation of the paper - Axiomatic Attribution for Deep Networks.
- MalConv-Pytorch: Pytorch implementation of MalConv.
- trellisnet: Trellis Networks for Sequence Modeling
- Learning to Communicate with Deep Multi-Agent Reinforcement Learning: pytorch implementation of Learning to Communicate with Deep Multi-Agent Reinforcement Learning paper.
- pnn.pytorch: PyTorch implementation of CVPR'18 - Perturbative Neural Networks http://xujuefei.com/pnn.html.
- Face_Attention_Network: Pytorch implementation of face attention network as described in Face Attention Network: An Effective Face Detector for the Occluded Faces.
- waveglow: A Flow-based Generative Network for Speech Synthesis.
- deepfloat: This repository contains the SystemVerilog RTL, C++, HLS (Intel FPGA OpenCL to wrap RTL code) and Python needed to reproduce the numerical results in "Rethinking floating point for deep learning"
- EPSR: Pytorch implementation of Analyzing Perception-Distortion Tradeoff using Enhanced Perceptual Super-resolution Network. This work has won the first place in PIRM2018-SR competition (region 1) held as part of the ECCV 2018.
- ClariNet: A Pytorch Implementation of ClariNet arxiv.org/abs/1807.07281
- pytorch-pretrained-BERT: PyTorch version of Google AI's BERT model with script to load Google's pre-trained models
- torch_waveglow: A PyTorch implementation of the WaveGlow: A Flow-based Generative Network for Speech Synthesis.
- 3DDFA: The pytorch improved re-implementation of TPAMI 2017 paper: Face Alignment in Full Pose Range: A 3D Total Solution.
- loss-landscape: loss-landscape Code for visualizing the loss landscape of neural nets.
- famos: Pytorch implementation of the paper "Copy the Old or Paint Anew? An Adversarial Framework for (non-) Parametric Image Stylization" available at http://arxiv.org/abs/1811.09236.
- back2future.pytorch: This is a Pytorch implementation of Janai, J., Güney, F., Ranjan, A., Black, M. and Geiger, A., Unsupervised Learning of Multi-Frame Optical Flow with Occlusions. ECCV 2018.
- FFTNet: Unofficial Implementation of FFTNet vocode paper.
- FaceBoxes.PyTorch: A PyTorch Implementation of FaceBoxes.
- Transformer-XL: Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contexthttps://github.com/kimiyoung/transformer-xl
- associative_compression_networks: Associative Compression Networks for Representation Learning.
- fluidnet_cxx: FluidNet re-written with ATen tensor lib.
- Deep-Reinforcement-Learning-Algorithms-with-PyTorch: This repository contains PyTorch implementations of deep reinforcement learning algorithms.
- Shufflenet-v2-Pytorch: This is a Pytorch implementation of faceplusplus's ShuffleNet-v2.
- GraphWaveletNeuralNetwork: This is a Pytorch implementation of Graph Wavelet Neural Network. ICLR 2019.
- AttentionWalk: This is a Pytorch implementation of Watch Your Step: Learning Node Embeddings via Graph Attention. NIPS 2018.
- SGCN: This is a Pytorch implementation of Signed Graph Convolutional Network. ICDM 2018.
- SINE: This is a Pytorch implementation of SINE: Scalable Incomplete Network Embedding. ICDM 2018.
- GAM: This is a Pytorch implementation of Graph Classification using Structural Attention. KDD 2018.
- neural-style-pt: A PyTorch implementation of Justin Johnson's Neural-style.
- TuckER: TuckER: Tensor Factorization for Knowledge Graph Completion.
- pytorch-prunes: Pruning neural networks: is it time to nip it in the bud?
- SimGNN: SimGNN: A Neural Network Approach to Fast Graph Similarity Computation.
- Character CNN: PyTorch implementation of the Character-level Convolutional Networks for Text Classification paper.
- XLM: PyTorch original implementation of Cross-lingual Language Model Pretraining.
- DiffAI: A provable defense against adversarial examples and library for building compatible PyTorch models.
- APPNP: Combining Neural Networks with Personalized PageRank for Classification on Graphs. ICLR 2019.
- NGCN: A Higher-Order Graph Convolutional Layer. NeurIPS 2018.
- gpt-2-Pytorch: Simple Text-Generator with OpenAI gpt-2 Pytorch Implementation
- Splitter: Splitter: Learning Node Representations that Capture Multiple Social Contexts. (WWW 2019).
- CapsGNN: Capsule Graph Neural Network. (ICLR 2019).
- BigGAN-PyTorch: The author's officially unofficial PyTorch BigGAN implementation.
- ppo_pytorch_cpp: This is an implementation of the proximal policy optimization algorithm for the C++ API of Pytorch.
- RandWireNN: Implementation of: "Exploring Randomly Wired Neural Networks for Image Recognition".
- Zero-shot Intent CapsNet: GPU-accelerated PyTorch implementation of "Zero-shot User Intent Detection via Capsule Neural Networks".
- SEAL-CI Semi-Supervised Graph Classification: A Hierarchical Graph Perspective. (WWW 2019).
- MixHop: MixHop: Higher-Order Graph Convolutional Architectures via Sparsified Neighborhood Mixing. ICML 2019.
- densebody_pytorch: PyTorch implementation of CloudWalk's recent paper DenseBody.
- voicefilter: Unofficial PyTorch implementation of Google AI's VoiceFilter system http://swpark.me/voicefilter.
- NVIDIA/semantic-segmentation: A PyTorch Implementation of Improving Semantic Segmentation via Video Propagation and Label Relaxation, In CVPR2019.
- ClusterGCN: A PyTorch implementation of "Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks" (KDD 2019).
- NVlabs/DG-Net: A PyTorch implementation of "Joint Discriminative and Generative Learning for Person Re-identification" (CVPR19 Oral).
- NCRF: Cancer metastasis detection with neural conditional random field (NCRF)
- pytorch-sift: PyTorch implementation of SIFT descriptor.
- brain-segmentation-pytorch: U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI.
- glow-pytorch: PyTorch implementation of Glow, Generative Flow with Invertible 1x1 Convolutions (arxiv.org/abs/1807.03039)
- EfficientNets-PyTorch: A PyTorch implementation of EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks.
- STEAL: STEAL - Learning Semantic Boundaries from Noisy Annotations nv-tlabs.github.io/STEAL
- EigenDamage-Pytorch: Official implementation of the ICML'19 paper "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis".
- Aspect-level-sentiment: Code and dataset for ACL2018 paper "Exploiting Document Knowledge for Aspect-level Sentiment Classification"
- breast_cancer_classifier: Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening arxiv.org/abs/1903.08297
- DGC-Net: A PyTorch implementation of "DGC-Net: Dense Geometric Correspondence Network".
- universal-triggers: Universal Adversarial Triggers for Attacking and Analyzing NLP (EMNLP 2019)
- Deep-Reinforcement-Learning-Algorithms-with-PyTorch: PyTorch implementations of deep reinforcement learning algorithms and environments.
- simple-effective-text-matching-pytorch: A pytorch implementation of the ACL2019 paper "Simple and Effective Text Matching with Richer Alignment Features".
- Adaptive-segmentation-mask-attack (ASMA): A pytorch implementation of the MICCAI2019 paper "Impact of Adversarial Examples on Deep Learning Models for Biomedical Image Segmentation".
- NVIDIA/unsupervised-video-interpolation: A PyTorch Implementation of Unsupervised Video Interpolation Using Cycle Consistency, In ICCV 2019.
- Seg-Uncertainty: Unsupervised Scene Adaptation with Memory Regularization in vivo, In IJCAI 2020.
- pulse: Self-Supervised Photo Upsampling via Latent Space Exploration of Generative Models
- distance-encoding: Distance-Encoding - Design Provably More PowerfulGNNs for Structural Representation Learning.
- Pathfinder Discovery Networks: Pathfinder Discovery Networks for Neural Message Passing.
- PyKEEN: A Python library for learning and evaluating knowledge graph embeddings.
- SSSNET: Official implementation of the SDM2022 paper "SSSNET: Semi-Supervised Signed Network Clustering".
- MagNet: Official implementation of the NeurIPS2021 paper "MagNet: A Neural Network for Directed Graphs".
- Semantic Search: Latest in the field of neural information retrieval / semantic search.
- FreeGrad - PyTorch library for custom backward passes, straight-through estimators and gradient transforms.
演讲与会议
- PyTorch 2018开发者大会:2018年首次举办的PyTorch开发者大会。
PyTorch相关资源
- the-incredible-pytorch:令人惊叹的PyTorch:一个精心整理的列表,包含与PyTorch相关的教程、论文、项目、社区等资源。
- 生成模型:在TensorFlow、Keras和PyTorch中实现的生成模型集合,如GAN、VAE等。http://wiseodd.github.io
- PyTorch与TensorFlow对比:Reddit上的一篇信息丰富的讨论帖。
- PyTorch讨论论坛
- PyTorch笔记本:Docker堆栈:一个类似于Jupyter Notebook Scientific Python Stack的项目。
- 像鲍勃·罗斯一样绘画:利用神经网络的力量,用PyTorch像鲍勃·罗斯一样作画!
- pytorch-tvmisc:适用于PyTorch的多功能杂项工具集。
- PyTorch A3C Mujoco实现:在Mujoco Gym环境中实现A3C算法。
- 5分钟学会PyTorch。
- PyTorch聊天机器人:使用PyTorch实现的一款出色的聊天机器人。
- Malmo协作AI挑战赛:Malmo协作AI挑战赛——猪捕手团队。
- sketchnet:一个能够接收图像并生成Processing源代码以重新生成该图像的模型。
- 深度学习训练营:由非营利社区运营的为期5天的深度学习训练营 http://deep-ml.com。
- 亚马逊森林计算机视觉项目:使用PyTorch/Keras编写的卫星图像标注代码,包含大量PyTorch技巧。参与Kaggle竞赛。
- AlphaZero五子棋实现:AlphaZero算法在五子棋(又称连珠或五子棋)中的实现。
- pytorch-cv:用于目标检测、分割和姿态估计的仓库。
- 深度人体重识别:基于PyTorch的深度人体重识别方法实现。
- PyTorch模板项目。
- 《使用PyTorch进行深度学习》图书:一本实用指南,介绍如何使用PyTorch构建文本和视觉领域的神经网络模型。亚马逊购买链接 GitHub代码库
- TensorFlow与PyTorch对比:比较用TensorFlow和PyTorch编写的层之间的输出结果。
- hasktorch:在Haskell中实现张量和神经网络。
- 《使用PyTorch进行深度学习》书籍:本书教你如何使用Python和PyTorch实现深度学习算法。
- nimtorch:PyTorch - Python + Nim。
- derplearning:自动驾驶遥控车代码。
- pytorch-saltnet:Kaggle TGS盐矿识别挑战赛中单模型解决方案第9名。
- PyTorch脚本:一些专为Windows系统设计的PyTorch脚本。
- pytorch_misc:为PyTorch讨论区创建的代码片段。
- awesome-pytorch-scholarship:一份优秀的PyTorch奖学金文章、指南、博客、课程及其他资源列表。
- MentisOculi:用PyTorch编写的光线追踪器(raynet?)。
- 涂鸦大师:“不要编写你的UI,而是画出来!”
- ocaml-torch:PyTorch的OCaml绑定。
- 扩展脚本:用于TorchScript自定义C++/CUDA算子的示例仓库。
- PyTorch推理:在Windows 10平台上使用C++实现PyTorch 1.0推理。
- PyTorch C++推理:以C++作为Web服务器来部署PyTorch 1.0模型。
- tch-rs:PyTorch的Rust绑定。
- TorchSharp:PyTorch引擎的.NET绑定。
- ML Workspace:面向机器学习和数据科学的一体化Web IDE。将Jupyter、VS Code、PyTorch等多种工具/库整合到一个Docker镜像中。
- PyTorch编码规范:PyTorch代码的编码规范。一致且良好的代码风格有助于协作并避免错误!
反馈:如果您有任何想法,或者希望在此列表中添加其他内容,请随时贡献。
常见问题
相似工具推荐
stable-diffusion-webui
stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。
everything-claude-code
everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上
ComfyUI
ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。
NextChat
NextChat 是一款轻量且极速的 AI 助手,旨在为用户提供流畅、跨平台的大模型交互体验。它完美解决了用户在多设备间切换时难以保持对话连续性,以及面对众多 AI 模型不知如何统一管理的痛点。无论是日常办公、学习辅助还是创意激发,NextChat 都能让用户随时随地通过网页、iOS、Android、Windows、MacOS 或 Linux 端无缝接入智能服务。 这款工具非常适合普通用户、学生、职场人士以及需要私有化部署的企业团队使用。对于开发者而言,它也提供了便捷的自托管方案,支持一键部署到 Vercel 或 Zeabur 等平台。 NextChat 的核心亮点在于其广泛的模型兼容性,原生支持 Claude、DeepSeek、GPT-4 及 Gemini Pro 等主流大模型,让用户在一个界面即可自由切换不同 AI 能力。此外,它还率先支持 MCP(Model Context Protocol)协议,增强了上下文处理能力。针对企业用户,NextChat 提供专业版解决方案,具备品牌定制、细粒度权限控制、内部知识库整合及安全审计等功能,满足公司对数据隐私和个性化管理的高标准要求。
ML-For-Beginners
ML-For-Beginners 是由微软推出的一套系统化机器学习入门课程,旨在帮助零基础用户轻松掌握经典机器学习知识。这套课程将学习路径规划为 12 周,包含 26 节精炼课程和 52 道配套测验,内容涵盖从基础概念到实际应用的完整流程,有效解决了初学者面对庞大知识体系时无从下手、缺乏结构化指导的痛点。 无论是希望转型的开发者、需要补充算法背景的研究人员,还是对人工智能充满好奇的普通爱好者,都能从中受益。课程不仅提供了清晰的理论讲解,还强调动手实践,让用户在循序渐进中建立扎实的技能基础。其独特的亮点在于强大的多语言支持,通过自动化机制提供了包括简体中文在内的 50 多种语言版本,极大地降低了全球不同背景用户的学习门槛。此外,项目采用开源协作模式,社区活跃且内容持续更新,确保学习者能获取前沿且准确的技术资讯。如果你正寻找一条清晰、友好且专业的机器学习入门之路,ML-For-Beginners 将是理想的起点。
ragflow
RAGFlow 是一款领先的开源检索增强生成(RAG)引擎,旨在为大语言模型构建更精准、可靠的上下文层。它巧妙地将前沿的 RAG 技术与智能体(Agent)能力相结合,不仅支持从各类文档中高效提取知识,还能让模型基于这些知识进行逻辑推理和任务执行。 在大模型应用中,幻觉问题和知识滞后是常见痛点。RAGFlow 通过深度解析复杂文档结构(如表格、图表及混合排版),显著提升了信息检索的准确度,从而有效减少模型“胡编乱造”的现象,确保回答既有据可依又具备时效性。其内置的智能体机制更进一步,使系统不仅能回答问题,还能自主规划步骤解决复杂问题。 这款工具特别适合开发者、企业技术团队以及 AI 研究人员使用。无论是希望快速搭建私有知识库问答系统,还是致力于探索大模型在垂直领域落地的创新者,都能从中受益。RAGFlow 提供了可视化的工作流编排界面和灵活的 API 接口,既降低了非算法背景用户的上手门槛,也满足了专业开发者对系统深度定制的需求。作为基于 Apache 2.0 协议开源的项目,它正成为连接通用大模型与行业专有知识之间的重要桥梁。