最佳PyTorch工具集合:AI开发者的实用资源

pytorch-AdaIN

pytorch-AdaIN

AdaIN风格迁移

PyTorch实现的实时风格迁移技术

CnSTD

CnSTD

场景文字检测数学公式检测

多语言场景文字与数学公式检测工具包

sequitur

sequitur

sequitur自编码器

高效创建和训练序列数据自编码器的Python库

pytorch-ts

pytorch-ts

PyTorchTS时间序列预测

概率时间序列预测开源框架

UnsupervisedScalableRepresentationLearningTimeSeries

UnsupervisedScalableRepresentationLearningTimeSeries

时间序列表示学习无监督学习

多变量时间序列的无监督可扩展表示学习方法

pytorch-llama

pytorch-llama

LLaMA 2PyTorch

基于PyTorch的LLaMA 2模型实现

mmpose

mmpose

MMPose姿态估计

先进的开源姿态估计工具箱

tianshou

tianshou

Tianshou强化学习

基于PyTorch的高性能模块化强化学习框架

GPT2

GPT2

GPT-2PyTorch

PyTorch优化实现的自然语言生成模型

recurrent-interface-network-pytorch

recurrent-interface-network-pytorch

RIN生成扩散模型

无需级联网络的高效图像视频生成模型

modular-diffusion

modular-diffusion

Modular Diffusion扩散模型

灵活可扩展的PyTorch扩散模型框架

ema-pytorch

ema-pytorch

EMAPyTorch

PyTorch模型指数移动平均跟踪工具

neurodiffeq

neurodiffeq

neurodiffeq神经网络

神经网络求解微分方程的开源Python库

pytorch-blender

pytorch-blender

blendtorchPyTorch

将Blender与PyTorch融合的深度学习框架

3DModelingRL

3DModelingRL

3D建模强化学习

深度强化学习在3D建模中的应用与突破

evotorch

evotorch

EvoTorch进化计算

基于PyTorch的高性能进化计算库

BasicSR

BasicSR

BasicSR图像复原

基于PyTorch的图像视频复原工具箱 实现多种先进算法

CrypTen

CrypTen

CrypTen隐私保护机器学习

基于PyTorch的隐私保护机器学习框架

all-in-one

all-in-one

音乐结构分析AI模型

全能音乐结构分析工具 自动识别节奏节拍和段落

auraloss

auraloss

PyTorch音频处理

专注音频的PyTorch损失函数库