tiny-cuda-nn

专注于快速训练和查询神经网络的开源框架

Tiny CUDA Neural Networks 深度学习 CUDA C++编程 GPU Github 开源项目

Tiny CUDA Neural Networks是一个紧凑、高效的开源框架，专注于快速训练和查询神经网络。它包含优化的多层感知器（MLP）和多分辨率哈希编码，并支持多种输入编码、损失函数和优化器。适用于NVIDIA GPU，通过C++/CUDA API和PyTorch扩展，助力高性能计算和深度学习项目。

访问官网

Github

介绍相关项目

nni - 可自动执行特征工程、神经架构搜索、超参数调优和深度学习的模型压缩

GithubNNI开源项目架构搜索模型压缩神经网络智能优化超参数调整

NNI提供一站式解决方案，支持自动化的特征工程、神经架构搜索、超参数调整和模型压缩。它兼容多种框架，并提供详尽的API、丰富的示例及全面的教程。适用于多种训练环境，包括本地、远程SSH服务器和Kubernetes，帮助推动开源社区的技术发展。

torch-conv-kan - 引入基于Kolmogorov-Arnold表示理论的高效卷积神经网络

CUDAConvolutional layersGithubKolmogorov-Arnold NetworksPyTorchTorchConv KAN开源项目

项目展示了使用PyTorch和CUDA加速的Kolmogorov-Arnold网络（KAN）模型的训练、验证和量化，支持MNIST、CIFAR、TinyImagenet和Imagenet1k数据集的性能评估。当前项目持续开发，已发布涉及ResNet、VGG、DenseNet、U-net等架构的新模型和预训练权重，适用于医疗图像分割和高效卷积神经网络的进一步研究和优化。

unet.cu - UNet扩散模型的高性能CUDA实现

CUDAGithubUNet卷积神经网络图像生成开源项目深度学习

这个开源项目使用纯C++/CUDA实现了UNet扩散模型训练框架,支持无条件扩散。框架包含线性层、组归一化、注意力等核心算子的GPU加速实现,重点优化3x3卷积。通过多次迭代提升CUDA kernel性能,训练速度达PyTorch的40%。项目展示了深度学习框架在GPU上的高效实现过程,为相关开发提供参考。

BitNetMCU - 低比特量化神经网络在微控制器上的高精度实现

BitNetMCUGithubMNIST数据集开源项目微控制器模型训练量化神经网络

BitNetMCU项目致力于在低端微控制器上实现高精度的低比特量化神经网络。通过优化训练和推理过程，该项目在仅2KB RAM和16KB Flash的CH32V003等微控制器上，实现了16x16 MNIST数据集超过99%的测试准确率，无需使用乘法指令。项目提供基于PyTorch的训练流程和ANSI-C实现的推理引擎，便于在不同微控制器上应用。

tiny-dream - 轻量级 Stable Diffusion C++ 推理库

C++GithubStable DiffusionTiny Dream图像生成嵌入式开源项目

Tiny Dream 是一个轻量级的 Stable Diffusion C++ 实现，仅包含头文件且无外部依赖。该库专注于提高 CPU 效率和减少内存占用，在普通硬件上即可高效运行。Tiny Dream 内存需求低，支持多种高级功能，并提供简洁的 C++ API，便于集成到现有项目中。

caffe - 一个用于深度学习的快速开放框架

BAIRBVLCCaffeGithub开源项目模型动物园深度学习框架

Caffe是由伯克利AI研究中心和社区贡献者开发的深度学习框架，强调高效表达、速度和模块化。用户可以通过项目网站获取详细信息，包括DIY深度学习教程、文档、参考模型和社区模型库。Caffe提供多种自定义版本，例如优化CPU和多节点支持的Intel Caffe、适用于AMD和Intel设备的OpenCL Caffe，以及Windows Caffe。社区用户可通过Gitter聊天和Google论坛进行交流，提交问题和建议。项目遵循BSD 2-Clause许可证，鼓励在研究中引用。

TensorRT - 优化深度学习推理的开源平台

CUDADockerGithubNVIDIAONNXTensorRT开源项目

NVIDIA TensorRT 开源软件提供插件和 ONNX 解析器的源码，展示 TensorRT 平台功能的示例应用。这些组件是 TensorRT GA 版本的一部分，并包含扩展和修复。用户可以轻松安装 TensorRT Python 包或根据构建指南编译。企业用户可使用 NVIDIA AI Enterprise 套件，并可加入 TensorRT 社区获取最新产品更新和最佳实践。

nntrainer - 设备端神经网络训练与个性化框架

GithubNNtrainer个性化嵌入式设备开源项目机器学习神经网络

NNtrainer是专为资源受限的嵌入式设备设计的开源神经网络训练框架。支持k-NN、神经网络和逻辑回归等多种机器学习算法，提供少样本学习、ResNet和VGG等任务示例。通过设备端微调实现模型个性化，高效利用有限资源。NNtrainer独特之处在于支持设备端完整训练流程，而非仅限于推理。这使得它在保护用户数据隐私的同时，能够实现个性化模型优化。框架已在Samsung Galaxy智能手机和Ubuntu PC上验证可用。

ffcv-imagenet - 高效ImageNet训练框架提升模型性能

GithubImageNetPyTorchResNetffcv开源项目深度学习

ffcv-imagenet是一个高效的ImageNet训练框架，采用单文件PyTorch脚本实现。该项目能在标准方法1/10的时间内达到相同精度，支持多GPU并行和多模型同时训练。框架提供丰富的配置选项，结合FFCV数据加载和优化训练流程，使研究人员能更快迭代实验并获得高质量模型。项目还包含多种预设配置，适用于不同的训练需求和硬件环境。

convnextv2-tiny-1k-224 - 基于全新框架节点，优化卷积网络的性能

ConvNeXt V2FCMAEGithubHuggingfaceImageNet卷积神经网络图像分类开源项目模型

ConvNeXt V2引入了全新的全卷积掩码自动编码器框架及全局响应归一化层，提升卷积网络在多种识别基准的表现，经过ImageNet-1K数据集微调，适合高精度图像分类任务及视觉识别应用。

相关项目

项目侧边栏1

项目侧边栏2

推荐项目

Project Cover

豆包MarsCode

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

Project Cover

AI写歌

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

Project Cover

白日梦AI

白日梦AI提供专注于AI视频生成的多样化功能，包括文生视频、动态画面和形象生成等，帮助用户快速上手，创造专业级内容。

Project Cover

有言AI

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

Project Cover

Kimi

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

Project Cover

讯飞绘镜

讯飞绘镜是一个支持从创意到完整视频创作的智能平台，用户可以快速生成视频素材并创作独特的音乐视频和故事。平台提供多样化的主题和精选作品，帮助用户探索创意灵感。

Project Cover

讯飞文书

讯飞文书依托讯飞星火大模型，为文书写作者提供从素材筹备到稿件撰写及审稿的全程支持。通过录音智记和以稿写稿等功能，满足事务性工作的高频需求，帮助撰稿人节省精力，提高效率，优化工作与生活。

Project Cover

阿里绘蛙

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

Project Cover

AIWritePaper论文写作

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

使用协议隐私政策广告服务

投诉举报邮箱: service@vectorlightyear.com

@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号