Project Icon

pytorch-toolbelt

专为PyTorch设计的Python库,提供高效研发和Kaggle竞赛所需的工具集

pytorch-toolbelt是一款专为PyTorch设计的Python库,提供高效研发和Kaggle竞赛所需的工具集。其功能包括灵活的编码器-解码器架构、多种模块(如CoordConv、SCSE、Hypercolumn等)、GPU友好的测试时增强(TTA)、大图像推理及常用方法,支持多种损失函数,并与Catalyst库无缝集成。这些工具旨在简化模型构建、优化和推理过程。

gluon-cv - 计算机视觉领域的深度学习模型工具包,支持PyTorch和MXNet框架
GithubGluonCV图像分类对象检测开源项目深度学习计算机视觉
GluonCV是一个面向工程师、研究人员和学生的计算机视觉深度学习工具包,支持快速原型设计。其主要功能包括可复现SOTA结果的训练脚本、对PyTorch和MXNet框架的支持、大量预训练模型,以及简化实现的API设计和社区支持。用户还可以通过AutoGluon执行图像分类和目标检测任务。
theseus - 构建适用于机器人和视觉应用的定制非线性优化层
GithubPyTorchTheseus开源项目机器人神经网络非线性优化
Theseus 是一个高效的通用库,专门用于在 PyTorch 中构建定制的非线性优化层,支持机器人和视觉问题中的端到端可微分架构。其特性包括二阶非线性优化器、线性求解器、向量化和 GPU 加速,有助于提高计算速度和内存使用效率。该库通过结合领域专用模型和神经网络模型,在保持计算梯度的同时优化 AI 模型,非常适合研究人员和开发者使用。
serve - 提高PyTorch模型服务效率和安全性的关键技术
GithubPyTorchTorchServe大规模模型安全性开源项目模型服务
TorchServe是一款高效灵活的平台,用于生产环境中PyTorch模型的部署和扩展。最新版本通过默认启用的令牌授权机制和增强的模型API控制,有效预防未授权API调用和恶意代码风险。此外,该平台还支持在不同环境(包括本地、云服务及各类硬件)中快速部署模型。
LibMTL - 基于PyTorch的多任务学习开源库,支持多种架构和优化策略
GithubLibMTLPyTorch多任务学习开源库开源项目算法
LibMTL是一个基于PyTorch的开源库,专为多任务学习(MTL)设计。它提供了一致的代码库和评估流程,支持多种架构和优化策略,涵盖多个领域的基准数据集。LibMTL采用模块化设计,允许用户灵活添加自定义组件或调整现有算法,方便开发新策略或应用于新场景。详尽的文档确保不同经验水平的开发者都能轻松使用。
lightning-bolts - PyTorch Lightning的深度学习组件扩展
BoltsDeepSparseGithubPyTorch LightningSparseMLTorch ORT开源项目
Lightning Bolts为PyTorch Lightning提供了多种扩展组件,包括回调和数据集,旨在加速训练和推理。它支持通过Torch ORT将模型转换为优化的ONNX图,以实现GPU加速训练;并通过SparseML在微调中引入稀疏性,提高推理性能。项目支持广泛的问题解决,并欢迎用户贡献通用组件。了解更多安装和使用信息,请访问官方文档和社区支持平台。
keras-tuner - 兼具易用性和可扩展性的超参数优化工具
GithubKerasTunerPython 3.8+TensorFlow 2.0+开源项目机器学习模型超参数优化
KerasTuner是一个便捷且可扩展的超参数优化工具,可以有效解决超参数搜索过程中遇到的问题。用户可以通过define-by-run语法轻松配置搜索空间,并使用贝叶斯优化、Hyperband和随机搜索算法找到模型的最佳参数值。该工具对研究人员十分友好,便于进行新搜索算法的实验。KerasTuner适用于Python 3.8+和TensorFlow 2.0+,并提供详细的开发者指南和API参考文档。
llms_tool - 多功能大语言模型训练测试工具包
GithubHuggingFaceRLHF分布式训练大语言模型开源项目预训练
llms_tool是一个基于HuggingFace的大语言模型工具包,支持多种模型的训练、测试和部署。它提供预训练、指令微调、奖励模型训练和RLHF等功能,支持全参数和低参数量训练。工具包集成WebUI和终端预测界面,以及DeepSpeed分布式训练。涵盖ChatGLM、LLaMA、Bloom等主流模型,提供多种训练方法和量化选项。
alfred - 多功能Python深度学习工具集
Githubalfred-py可视化命令行工具开源项目深度学习部署
alfred-py是一个专为深度学习设计的Python工具库,提供数据可视化、命令行工具、模型部署等多种功能。通过导入模块或终端命令,可便捷使用其丰富的工具集。该库旨在提升深度学习开发效率,是一个功能全面的实用工具。
torchlm - 面向人脸关键点检测的开源工具包
Githubtorchlm人脸关键点检测开源项目数据增强模型训练深度学习
torchlm是一个开源的人脸关键点检测工具包,提供训练、评估、导出和推理功能。它包含100多种数据增强方法,支持30多种原生关键点增强,可与torchvision和albumentations集成。torchlm实现了PIPNet等先进模型,在多个基准数据集上性能出色。该项目简化了人脸关键点检测的开发流程,适用于研究和实际应用。
pytorch-frame - 模块化深度学习框架用于异构表格数据
GithubPyTorch Frame开源项目模块化框架深度学习神经网络表格数据
PyTorch Frame是一个为异构表格数据设计的深度学习框架,支持数值、分类、时间、文本和图像等多种列类型。它采用模块化架构,实现了先进的深度表格模型,并可与大型语言模型集成。该框架提供了便捷的mini-batch加载器、基准数据集和自定义数据接口,简化了表格数据的深度学习研究过程,适用于各层次研究人员。框架内置多个预实现的深度表格模型,如Trompt、FTTransformer和TabNet等,并提供与XGBoost等GBDT模型的性能对比基准。PyTorch Frame无缝集成于PyTorch生态系统,便于与其他PyTorch库协同使用,为端到端的深度学习研究提供了便利。
项目侧边栏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号