Project Icon

lightning-bolts

PyTorch Lightning的深度学习组件扩展

Lightning Bolts为PyTorch Lightning提供了多种扩展组件,包括回调和数据集,旨在加速训练和推理。它支持通过Torch ORT将模型转换为优化的ONNX图,以实现GPU加速训练;并通过SparseML在微调中引入稀疏性,提高推理性能。项目支持广泛的问题解决,并欢迎用户贡献通用组件。了解更多安装和使用信息,请访问官方文档和社区支持平台。

SDXL-Lightning - 快速高质量文本到图像生成的突破性模型
AI绘图GithubHuggingfaceSDXL-Lightning开源项目快速推理扩散模型文本生成图像模型
SDXL-Lightning是一个基于稳定扩散XL的文本到图像生成模型,能在几步内生成1024px高质量图像。它提供1步、2步、4步和8步蒸馏模型,其中2步、4步和8步模型在保持高速的同时,展现出优秀的生成质量。该项目开源了完整的UNet和LoRA检查点,方便用户在各种场景中应用。
pytorch_geometric - 图形神经网络开发库
GithubPyTorch Geometric图神经网络开源项目数据处理机器学习深度学习
PyTorch Geometric是一个基于PyTorch的图形神经网络库,旨在简化结构化数据的建模与训练流程。支持小批量和大规模图的处理,并提供全面的GPU加速、数据管道处理以及常用基准数据集。这使得它成为机器学习研究者和初学者理想的选择。
fast-bert - 快速训练和部署BERT与XLNet文本分类模型的深度学习库
Fast-BertGithub开源项目文本分类深度学习自然语言处理预训练模型
fast-bert是一个深度学习库,用于训练和部署基于BERT和XLNet的文本分类模型。它支持多类和多标签分类,提供数据处理、模型训练、参数调优和部署功能。该库集成了LAMB优化器和学习率查找器,旨在简化最新自然语言处理技术的应用过程。fast-bert适用于各类文本分类任务,能够帮助开发者快速构建高性能模型。
DI-hpc - 高性能计算组件加速强化学习算法
CUDADI-HPCGithubPyTorch开源项目强化学习算法加速
DI-HPC是一款专为强化学习算法设计的高性能计算组件,主要用于加速GAE、n-step TD和LSTM等常见模块。该组件支持前向和反向传播,适用于训练、数据收集和测试环节。DI-HPC兼容CUDA环境和多个PyTorch版本,提供简便的安装方式和性能测试工具。通过提升计算效率,DI-HPC为强化学习研究和开发提供了有力支持。
Py-Boost - Python实现的GPU加速梯度提升决策树库
GPU加速GithubONNX兼容Python库多输出训练开源项目梯度提升
Py-Boost是一个Python实现的GPU加速梯度提升决策树库。该项目提供简洁接口,支持GPU训练和推理,易于定制。特色功能包括SketchBoost算法高效处理多输出任务,以及ONNX格式支持。Py-Boost为研究和开发人员提供了探索梯度提升方法的灵活工具,同时保持了较高的运行效率。
dlwpt-code - 深入浅出PyTorch深度学习指南
Deep Learning with PyTorchGithubPyTorch开源项目机器学习深度学习编程
《Deep Learning with PyTorch》通过实际项目展示深度学习的基础知识,适合希望掌握PyTorch的开发者、计算机科学家、数据科学家及相关专业学生。书中提供了对深度学习的直观理解,并深入探讨PyTorch的部分功能,适合具备编程基础的读者。作者团队拥有丰富的实践经验和开源项目贡献,确保内容实用且前沿。
zeta - 通过模块化组件提升AI模型开发速度
GithubZeta人工智能模型开源项目模块化神经网络高性能
Zeta项目提供模块化、高性能和可扩展的构建块,使AI模型开发速度提高80%。该项目的功能模块包括Flash Attention、SwiGLU激活函数和RelativePositionBias,这些组件大幅提升了模型的效率和性能。Zeta专注于可用性、模块化和性能,已被广泛应用于数百个模型中。用户可以通过简单的安装步骤,快速开始模型的原型设计、训练和优化。
hummingbird - 用于将经过训练的传统 ML 模型编译为张量计算的库
GithubHummingbirdPyTorch加速推理开源项目机器学习模型神经网络
Hummingbird通过将训练好的传统机器学习模型编译为张量计算,使其能够借助神经网络框架(如PyTorch)加速。用户可利用神经网络框架的优化和硬件加速,无需重新设计模型。支持多种模型如决策树、随机森林、LightGBM和XGBoost,并提供易于替换的推理API。支持转换为PyTorch、TorchScript、ONNX和TVM格式。
accelerate - 简化多设备PyTorch训练的框架
AccelerateGithubPyTorch分布式训练开源项目混合精度设备管理
Accelerate是一个轻量级PyTorch训练框架,允许在CPU、GPU、TPU等多种设备上运行原生PyTorch脚本。它自动处理设备分配和混合精度训练,简化了分布式训练流程。研究人员和开发者可专注于模型开发,无需关注底层实现细节,从而加速AI模型的训练和部署。
caffe2 - 轻量级、模块化和可扩展的深度学习框架
Caffe2GithubPyTorch开源项目模块化深度学习框架高性能
Caffe2是一个以表达力、速度和模块化为设计理念的轻量级、模块化和可扩展的深度学习框架。欲了解更多信息,请访问caffe2.ai。
项目侧边栏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号