最佳PyTorch工具集合：AI开发者的实用资源
modded-nanogpt
NanoGPTPyTorch
基于PyTorch的高效GPT-2训练器变体
depyf
depyfPyTorch
深入解析PyTorch编译器工作原理的开源工具
Adan
Adan优化器
快速优化深度学习模型的新方法
taichi-nerfs
NeRFTaichi
基于Taichi和PyTorch的神经辐射场快速训练与渲染框架
CompressAI
CompressAI深度学习
基于PyTorch的端到端压缩研究开源库
pytorch-lr-finder
PyTorch学习率
PyTorch学习率范围测试工具
vissl
VISSL自监督学习
自监督视觉学习框架 促进计算机视觉研究
Pretrained-Language-Model
预训练语言模型自然语言处理
先进预训练语言模型与优化技术集合
apex
ApexPyTorch
NVIDIA Apex加速PyTorch混合精度与分布式训练
torchtune
torchtunePyTorch
PyTorch原生库助力简化大语言模型开发
gemma_pytorch
GemmaPyTorch
Gemma模型的官方PyTorch实现及多平台推理支持
torchcrepe
音高跟踪PyTorch
PyTorch实现的CREPE音高追踪算法
s2fft
S2FFT球谐变换
基于JAX和PyTorch的球谐变换Python库
sssegmentation
语义分割深度学习
开源语义分割工具箱 集成多种先进算法和模型
edm2
EDM2扩散模型
优化扩散模型训练动态的创新技术
segment-anything-fast
Segment AnythingPyTorch
高性能图像分割模型加速框架
open-metric-learning
open-metric-learning度量学习
开源的PyTorch度量学习框架 支持多模态嵌入训练
SLiMe
SLiMe图像分割
基于Stable Diffusion的单样本图像分割方法
nanoGPT
nanoGPTGPT训练
简洁高效的中型GPT模型训练框架
torchexplorer
TorchExplorer神经网络
交互式PyTorch模型结构和训练过程可视化工具
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号