最佳PyTorch工具集合：AI开发者的实用资源
stylegan2-ada-pytorch
StyleGAN2-ADAGAN
针对小数据集优化的StyleGAN2实现框架
opensphere
OpenSphere人脸识别
统一的高维球面人脸识别训练评估框架
geoopt
geooptPyTorch
黎曼流形优化的PyTorch扩展库
transformer_latent_diffusion
TransformerLatent Diffusion
基于 PyTorch 的 Transformer 潜在扩散文本生图模型
zennit
Zennit神经网络解释
基于PyTorch的神经网络解释与探索框架
PyTorch_Tutorial
PyTorch深度学习
PyTorch深度学习实践教程
pytorch-hed
HED边缘检测
PyTorch重实现的全息嵌套边缘检测HED算法
veScale
PyTorchLLM训练框架
基于PyTorch的大规模语言模型训练框架
ffcv-imagenet
ImageNetffcv
高效ImageNet训练框架提升模型性能
how-to-read-pytorch
PyTorch深度学习
通俗易懂的PyTorch核心概念教程 从张量运算到数据加载的全面指南
flash-attention-minimal
Flash AttentionCUDA
简化Flash Attention的CUDA和PyTorch最小化实现
sk2torch
sk2torchPyTorch
实现scikit-learn模型到PyTorch模块的转换
CC6204
Deep Learning神经网络
西语深度学习实战课程
pytorch-minimize
PyTorch优化
PyTorch多变量函数优化工具集
torch2trt
torch2trtTensorRT
PyTorch模型转TensorRT加速工具
plm-nlp-code
自然语言处理预训练模型
自然语言处理预训练模型代码实现与学习资源
xformers
xFormersTransformer
Transformer 研究加速工具
torch-imle
I-MLE深度学习
将离散优化算法融入深度学习的创新方法
nnsight
nnsight深度学习模型
解释和操作深度学习模型内部的Python包
bayesian-torch
Bayesian-Torch不确定性估计
贝叶斯神经网络层和不确定性估计的PyTorch扩展库
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号