Project Icon

tensordict

PyTorch张量集合操作的高效字典类工具

TensorDict是一个继承张量属性的字典类,为PyTorch提供简洁的张量集合操作方法。它支持异步设备传输、节点间快速通信、张量形状操作和分布式计算,提高了代码的可读性、紧凑性和模块化。这个工具适用于模型训练、优化器实现等机器学习任务,能有效简化代码结构,提升开发效率。

torchio - 深度学习医疗图像处理工具集
GithubPythonTorchIO医药图像开源项目数据增强深度学习
此工具集为深度学习医疗图像处理提供高效解决方案,涵盖读取、预处理、采样、增强和写入3D医疗图像等功能。支持多种图像转换操作,包括随机仿射变换和特定领域伪影模拟。受NiftyNet启发,该项目广泛应用于医学AI研究,提升数据处理效率和模型性能。
Ensemble-Pytorch - PyTorch集成学习框架助力模型优化
Ensemble-PyTorchGithubpytorch开源项目机器学习模型集成深度学习
Ensemble-Pytorch是一个为PyTorch设计的集成学习框架,旨在提高深度学习模型的性能和鲁棒性。该框架支持多种集成策略,如Fusion、Voting、Bagging和Gradient Boosting,适用于分类和回归任务。作为PyTorch生态系统的一部分,Ensemble-Pytorch提供简洁的API和详细文档,便于研究人员和开发者实现和优化集成模型。
dlpack - 促进深度学习框架间张量共享与协作
DLPackGithub内存共享开源项目张量结构深度学习框架
DLPack是一种开放的内存张量结构,用于深度学习框架间的张量共享。它简化了框架间的运算符共享,便于封装供应商级运算符实现,支持快速切换后端实现。作为跨框架复用的桥梁,DLPack不直接实现张量和操作,而是促进深度学习生态系统的协作,为用户提供更多运算符选择和框架混合使用的可能性。
torchchat - 使用Python和C/C++运行大型语言模型的轻量化实现
GithubPyTorchPythontorchchat大语言模型开源项目模型部署
torchchat可以通过Python和C/C++应用程序无缝运行大型语言模型(LLMs),支持桌面、服务器以及iOS和Android设备。该项目特点包括PyTorch原生执行、高效运行、支持多种硬件和操作系统、多种数据类型和量化方案。其安装步骤简便,并提供多种运行模式,如命令行、浏览器界面和REST API,适用于各类开发环境。
attorch - 易于修改的Python神经网络模块
GithubPyTorchTritonattorch开源项目深度学习神经网络模块
attorch是一个基于OpenAI Triton的PyTorch模块子集,提供易于修改的高效神经网络模块。支持自动混合精度、计算机视觉和自然语言处理相关层。
darts - 使用DARTS算法高效设计图像分类和语言建模架构
DARTSGithubPyTorch卷积架构图像分类开源项目语言建模
DARTS算法通过连续松弛和梯度下降,在架构空间中高效设计用于图像分类(CIFAR-10和ImageNet)和语言建模(Penn Treebank和WikiText-2)的高性能卷积和循环架构。只需一块GPU即可运行,提供预训练模型及详细的架构搜索和评估指南,支持自定义架构的可视化。
executorch - 移动和边缘设备上高效运行PyTorch模型的解决方案
ExecuTorchGithubPyTorch开源项目推理能力模型部署边缘设备
ExecuTorch 提供端到端解决方案,实现移动和边缘设备上的推理能力,涵盖穿戴设备、嵌入式设备和微控制器。作为 PyTorch Edge 生态系统的一部分,ExecuTorch 通过轻量级运行时,利用硬件能力(如 CPU、NPU 和 DSP),高效地将 PyTorch 模型部署到多种平台。其主要优势包括:广泛的兼容性、开发效率和出色的用户体验。欲了解更多技术细节和教程,请访问文档网站获取最新版本。
torchshow - 一行代码实现计算机视觉数据可视化
GithubTorchShow图像处理开源项目张量可视化计算机视觉调试工具
TorchShow是一个为计算机视觉项目开发的开源可视化工具。它能自动识别并处理各种张量类型,包括RGB图像、灰度图像、掩码等。支持单张和批量图像、视频、光流等多种数据的可视化。TorchShow以一行代码实现数据可视化的简洁特性,显著提升了项目开发和调试效率。适用于需要频繁验证张量数据的研究人员和开发者。
torchmd-net - 神经网络势能模型的高效训练与实现框架
GPU加速GithubPyTorchTorchMD-NET分子动力学开源项目神经网络势能
TorchMD-NET是一个先进的神经网络势能(NNP)模型框架,提供高效快速的NNP实现。该框架与ACEMD、OpenMM和TorchMD等GPU加速分子动力学代码集成,并将NNP作为PyTorch模块提供。项目支持等变Transformer、Transformer、图神经网络和TensorNet等多种架构,可通过conda-forge安装或从源代码构建。TorchMD-NET具有灵活的训练配置选项,支持自定义数据集和多节点训练,并提供预训练模型。
tensorboardX - TensorBoard可视化的轻量级Python扩展库
GithubTensorBoardtensorboardX可视化开源项目机器学习深度学习
tensorboardX是一个开源的轻量级Python库,用于简化TensorBoard可视化工具的使用。它支持多种数据类型的可视化,如标量、图像和音频等。通过简单的函数调用,开发者可以轻松记录实验数据。该库兼容PyTorch等多种深度学习框架,方便跟踪和分析机器学习实验。此外,tensorboardX还能与Comet平台集成,提供额外的实验管理功能。它具有良好的版本兼容性,为不同环境下的机器学习项目提供了灵活的可视化解决方案。
项目侧边栏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号