最佳PyTorch工具集合：AI开发者的实用资源
PiPPy
PiPPypipeline parallelism
PyTorch模型自动化管道并行工具
torchmd
TorchMD分子动力学
开源分子动力学模拟框架
uni2ts
Uni2TS时间序列预测
时间序列预测Transformer模型的统一训练框架
OpenTAD
OpenTAD时序动作检测
多功能时序动作检测工具箱支持多数据集和前沿方法
test-time-adaptation
在线测试时适应计算机视觉
多场景计算机视觉模型在线测试时适应框架
streaming
StreamingDataset云存储
面向大规模机器学习的高效数据流式加载工具
diffusion-models-class
Hugging Face扩散模型
掌握扩散模型从理论到实践的全面课程
benchmark
PyTorch基准测试
开源基准测试集评估PyTorch性能
k-diffusion
k-diffusion生成模型
扩散模型框架支持多种采样算法和模型架构
PPO-for-Beginners
PPO强化学习
PyTorch实现近端策略优化算法详解
muse-maskgit-pytorch
MuseAI绘图
基于掩码生成变压器的PyTorch文本到图像生成框架
DI-treetensor
DI-treetensor张量
树状张量结构简化深度学习中的复杂计算
terratorch
TerraTorch地理空间数据
强大灵活的地理空间基础模型微调框架
lightning-uq-box
Lightning-UQ-Box不确定性量化
神经网络不确定性量化开源工具库
Knowledge-Distillation-Toolkit
知识蒸馏模型压缩
开源知识蒸馏工具包助力机器学习模型压缩
torchmd-net
TorchMD-NET神经网络势能
神经网络势能模型的高效训练与实现框架
LFM
Flow Matching潜在空间
潜空间流匹配实现高效图像生成
fast-DiT
DiT扩散模型
改进PyTorch实现的可扩展扩散模型转换器
mlp
多层感知器自然语言模型
多层感知器实现n-gram语言模型的开源项目
fvcore
fvcore计算机视觉
FAIR开发的轻量级计算机视觉库 提供核心共享功能
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号