Project Icon

torchrec

旨在提供大规模推荐系统所需的常见稀疏性和并行性原语的PyTorch库

TorchRec是一个专为大规模推荐系统设计的PyTorch库,提供稀疏性和并行性解决方案。它支持多种嵌入表分片策略,并能自动优化分片计划。通过流水线训练和优化内核,提高模型性能。还支持量化训练和推理,包含多个验证的模型架构和数据集示例,适用于需要高性能和扩展性的推荐系统项目。

UnboundedNeRFPytorch - 大规模神经辐射场基准测试的指南
BenchmarkGithubNeRFPytorchState-of-the-artUnbounded Neural Radiance Fields开源项目
UnboundedNeRFPytorch项目专注于基准测试多种最新的大规模神经辐射场(NeRF)算法,并提供简洁高效的代码库。项目展示了在Unbounded Tanks & Temples和Mip-NeRF-360基准测试中的优秀表现,旨在帮助研究人员和开发者提升NeRF应用效果。包括详细的安装步骤、数据处理指南和训练自定义NeRF模型的方法,适合技术用户快速上手并获得佳绩。
recsys-dataset - OTTO电商会话推荐系统大规模开放数据集
Github会话数据集多目标优化开源项目推荐系统电子商务评估指标
OTTO开放的电商推荐系统数据集包含1200万匿名用户会话和2.2亿次交互事件,涵盖180万商品。数据以.jsonl格式提供,便于研究人员直接使用。该数据集专为多目标和基于会话的推荐系统研究设计,定义了相应的评估指标,可作为该领域的基准数据集。数据集来源于OTTO真实电商平台,包括用户点击、加购和下单行为。研究人员可利用此数据集开发和评估多目标推荐算法,尤其适合基于会话的推荐系统研究。该数据集的开放将促进电商推荐系统领域的学术研究和技术创新。
torch-points3d - 用于在点云上进行深度学习的 Pytorch 框架
CUDAGithubPyTorchtorch-points3d开源项目深度学习点云分析
一个用于点云分析的深度学习框架,基于Pytorch Geometric和Facebook Hydra。该框架支持构建复杂模型并提供高层次API,支持PointNet、PointNet++、RSConv等常见模型,便捷实现分类、分割和检测任务。推荐使用Docker安装以确保兼容性。了解更多信息,请查阅文档和示例笔记本。
executorch - 移动和边缘设备上高效运行PyTorch模型的解决方案
ExecuTorchGithubPyTorch开源项目推理能力模型部署边缘设备
ExecuTorch 提供端到端解决方案,实现移动和边缘设备上的推理能力,涵盖穿戴设备、嵌入式设备和微控制器。作为 PyTorch Edge 生态系统的一部分,ExecuTorch 通过轻量级运行时,利用硬件能力(如 CPU、NPU 和 DSP),高效地将 PyTorch 模型部署到多种平台。其主要优势包括:广泛的兼容性、开发效率和出色的用户体验。欲了解更多技术细节和教程,请访问文档网站获取最新版本。
gemma_pytorch - Gemma模型的官方PyTorch实现及多平台推理支持
AI模型GemmaGithubPyTorch开源项目机器学习自然语言处理
gemma_pytorch项目是Gemma模型的官方PyTorch实现,支持CPU、GPU和TPU多平台推理。项目提供PyTorch和PyTorch/XLA两种实现,涵盖2B、7B、9B和27B等多个模型变体及量化版本。通过Docker可快速部署环境并进行推理。项目还包含最新的Gemma v2和CodeGemma模型支持,为用户提供全面的Gemma模型应用方案。
raft - 可复用CUDA函数库加速向量搜索与机器学习
CUDAGPU加速GithubRAFT向量搜索开源项目机器学习
RAFT是一个CUDA加速的C++头文件库,为机器学习和信息检索提供基础算法和原语。它实现了先进的向量相似度搜索方法,包括暴力搜索、IVF-Flat、IVF-PQ和CAGRA。RAFT还提供可重用原语,用于构建涵盖数据生成、模型评估、分类回归、聚类等领域的机器学习算法。通过跨项目复用和集中核心计算,RAFT加速了算法开发,并使未来优化能广泛应用于各种算法。
open-metric-learning - 开源的PyTorch度量学习框架 支持多模态嵌入训练
GithubPyTorchopen-metric-learning嵌入向量度量学习开源项目检索系统
open-metric-learning是一个基于PyTorch的开源度量学习框架,用于训练和验证高质量嵌入模型。它提供端到端流水线、实用案例和预训练模型库,支持图像和文本等多种模态。该框架具有统一的检索结果处理和评估方法,适用于人脸识别、商品搜索等嵌入学习任务。已被多家知名公司和机构采用,是一个功能丰富、易于上手的度量学习工具。
relora - 使用ReLoRA实现高效深度学习模型训练
GithubPEFTPyTorchReLoRAflash attention开源项目训练
ReLoRA项目通过低秩更新实现高效深度学习训练,兼容Python 3.10+和PyTorch 2.0+,支持更大的微批次大小。用户可通过执行预处理和不同配置的训练脚本,达到高效分布式训练,并支持cosine重启调度和多GPU配置。项目涵盖了预热训练和ReLoRA重启,适用于各种规模的模型训练。
pytorch-ts - 概率时间序列预测开源框架
GithubPyTorchPyTorchTS开源项目时间序列预测概率模型深度学习
PyTorchTS是一个基于PyTorch的开源时间序列预测框架,利用GluonTS作为后端API。它提供先进的概率模型,支持数据处理和回测。该框架适用于单变量和多变量时间序列预测,安装简便,易于使用。PyTorchTS为数据科学家和研究人员提供了高效的时间序列分析工具。
项目侧边栏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号