最佳PyTorch工具集合：AI开发者的实用资源
cycle-diffusion
CycleDiffusion扩散模型
零样本图像翻译与无配对图片转换的扩散模型方法
keras_cv_attention_models
Keras_cv_attention_modelsTensorFlow
深度学习模型和使用指南
stable-fast
Stable FastDiffusers
优化HuggingFace Diffusers推理性能的轻量级框架
onediff
onediffHF diffusers
扩散模型加速库
InfiniTransformer
InfiniTransformerPyTorch
非官方的PyTorch和Transformers实现
daam
DAAMStable Diffusion
解释稳定扩散模型的跨注意力归因图方法
skorch
skorchPyTorch
scikit-learn兼容的PyTorch神经网络库
fairseq
Fairseq序列建模
序列建模工具包，支持机器翻译与文本生成
DragGAN
DragGANStyleGAN3
基于交互点的生成图像操作
cramming
Cramming Language ModelBERT
探索单GPU一天内训练BERT语言模型的极限
prismer
PrismerPrismerZ
集成多任务专家的先进视觉语言模型
pytorch-cpp
PyTorchC++
C++ 实现的 PyTorch 教程，为深度学习研究者提供从基础到高级的全面指南
open_flamingo
OpenFlamingoVision-Language Model
开源多任务视觉语言模型，支持图像文本生成和多模态训练
SparseBEV
SparseBEV立体检测
多摄像头视频中的高性能稀疏3D目标检测技术
hardware-aware-transformers
HATTransformer
瞄准多硬件平台优化的自然语言处理Transformer模型
TextPruner
TextPruner模型剪枝
使用低成本且无需训练的方法优化预训练语言 模型
TransMorph_Transformer_for_Medical_Image_Registration
TransMorphTransformer
基于Transformer的无监督医学图像配准方法
ao
torchaoPyTorch
优化PyTorch工作流，实现高性能和内存占用减少
open-cd
Open-CD开源
开源变更检测工具箱，支持多种视觉任务模型
exporters
Core MLTransformers
将Transformer模型高效转换为Core ML格式
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号