Project Icon

depyf

深入解析PyTorch编译器工作原理的开源工具

depyf是一个开源工具,专注于解析PyTorch编译器torch.compile的内部工作机制。这个工具支持多个Python版本,提供简洁的API接口,能够生成详细的编译过程文档,并支持代码调试。通过depyf,机器学习研究人员和开发者可以深入理解编译过程,适应编译机制,并优化代码以提升性能。作为PyTorch生态系统的一员,depyf为用户提供了深入了解PyTorch编译器的机会。

d2l-pytorch - MXNet代码转换为PyTorch实现的指南
Dive Into Deep LearningGithubPyTorch卷积神经网络开源项目深度学习线性神经网络
本项目基于《Dive Into Deep Learning》书籍,将MXNet代码转换为PyTorch实现。内容包括安装指南、线性神经网络、多层感知器、卷积神经网络、现代卷积网络、循环神经网络和注意力机制等章节。提供详细教程和示例代码,适合使用PyTorch进行深度学习的开发者。建议克隆仓库或使用nbviewer查看notebook文件。
annotated_deep_learning_paper_implementations - 简洁易懂的PyTorch神经网络和算法实现
GANGithubPyTorchReinforcement LearningTransformerlabml.ai开源项目
该项目提供详细文档和解释的简明PyTorch神经网络及算法实现,涵盖Transformer、GPT-NeoX、GAN、扩散模型等前沿领域,并每周更新新实现,帮助研究者和开发者高效理解深度学习算法。
openfold - 增强蛋白质结构预测功能的AlphaFold2 PyTorch复现版本
AlphaFold 2DeepMindGithubOpenFoldPyTorch开源项目蛋白质结构预测
OpenFold是DeepMind AlphaFold 2的可训练PyTorch复现版本,提供高效的蛋白质结构预测解决方案。详细的安装、模型推理和训练指南可在文档主页找到。项目采用Apache Licence 2.0许可,使用的DeepMind预训练参数遵循CC BY 4.0许可。欢迎社区通过提交问题或拉请求进行贡献。引用OpenFold时应同时参考相关的AlphaFold研究成果。
fvcore - FAIR开发的轻量级计算机视觉库 提供核心共享功能
FAIRGithubPyTorchfvcore开源项目深度学习计算机视觉
fvcore是FAIR开发的轻量级计算机视觉库,为多个框架提供核心共享功能。它包含常用PyTorch组件、FLOP计数工具、参数计数、BatchNorm统计重计算和超参数调度器等特性。该库支持Detectron2、PySlowFast和ClassyVision等项目,所有组件经过严格测试,兼容Python 3.6+和PyTorch环境。
pytorch-dnc - PyTorch实现的差分神经计算机及相关模型库
DNCGithubSAMSDNC开源项目神经网络记忆增强
这个PyTorch库实现了差分神经计算机(DNC)、稀疏访问存储器(SAM)和稀疏差分神经计算机(SDNC)等模型。它提供灵活API用于构建和训练这些神经网络,支持多层控制器、共享内存等配置。库中还包含复制和加法等基准任务,以及内存可视化功能,有助于开发和评估基于外部存储的神经网络模型。
pythia-6.9b - 促进大规模语言模型可解释性研究的开源工具
GithubHuggingfacePythia大语言模型开源项目机器学习模型模型训练自然语言处理
Pythia-6.9b是EleutherAI开发的开源大规模语言模型,旨在促进可解释性研究。该模型包含69亿参数,在Pile数据集上训练,提供154个中间检查点,便于研究人员探索模型训练过程。Pythia-6.9b采用Transformer架构,性能与同类模型相当,主要用于学术研究而非直接部署。模型可能存在偏见风险,研究人员可通过Hugging Face Transformers库轻松使用。Pythia-6.9b为语言模型研究提供了理想的实验平台。
stable-fast - 优化HuggingFace Diffusers推理性能的轻量级框架
CUDADiffusersGithubPyTorchStable FastTorchScript开源项目
stable-fast是一个优化HuggingFace Diffusers推理性能的轻量级框架,支持NVIDIA GPU。相比TensorRT和AITemplate需要几十分钟的编译时间,stable-fast仅需几秒钟即可完成模型编译。主要特色包括动态形状、低精度计算和多种算子融合。它还兼容ControlNet和LoRA,并支持最新的StableVideoDiffusionPipeline,是加速PyTorch推理的有效工具。
pyg-lib - 图形神经网络高性能计算库
CUDAGithubPyTorchPythonpyg-lib安装开源项目
pyg-lib是一款专为图形神经网络优化的高性能计算库。该项目为Linux、Windows和macOS等主流操作系统提供预构建的Python包,兼容多个PyTorch版本和CUDA组合,支持Python 3.8至3.12。pyg-lib通过提升图形神经网络的计算效率,为研究和开发提供了实用工具。研究人员和开发者可根据具体的系统环境,使用pip命令快速安装所需版本,轻松增强图形神经网络的性能。
pytorch-cpp - C++ 实现的 PyTorch 教程,为深度学习研究者提供从基础到高级的全面指南
C++GithubLibTorchPyTorch开源项目教程深度学习
本项目提供了 C++ 版本的 PyTorch 教程,适用于从基础到高级的深度学习研究者,涵盖线性回归、卷积神经网络和生成对抗网络等内容。支持 macOS、Linux 和 Windows 的多平台编译和运行,项目要求包括 C++-17 兼容编译器、CMake 和合适版本的 LibTorch。含有全面的构建与运行指南,以及交互式教程和 Docker 支持。
项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号