torchmd-net

神经网络势能模型的高效训练与实现框架

TorchMD-NET 神经网络势能分子动力学 PyTorch GPU加速 Github 开源项目

TorchMD-NET是一个先进的神经网络势能(NNP)模型框架，提供高效快速的NNP实现。该框架与ACEMD、OpenMM和TorchMD等GPU加速分子动力学代码集成，并将NNP作为PyTorch模块提供。项目支持等变Transformer、Transformer、图神经网络和TensorNet等多种架构，可通过conda-forge安装或从源代码构建。TorchMD-NET具有灵活的训练配置选项，支持自定义数据集和多节点训练，并提供预训练模型。

访问官网

Github

Huggingface

介绍相关项目

Megatron-LM - 优化GPU训练技术加速大规模Transformer模型

GPU优化GithubMegatron-CoreMegatron-LM分布式训练大语言模型开源项目

Megatron-LM框架利用GPU优化技术实现Transformer模型的大规模训练。其Megatron-Core组件提供模块化API和系统优化,支持自定义模型训练。该项目可进行BERT、GPT、T5等模型预训练,支持数千GPU分布式训练百亿参数级模型,并提供数据预处理、模型评估和下游任务功能。

MT-UNet - 融合Transformer和UNet的医学图像分割新模型

GithubMT-UNet医学图像分割开源项目数据集准备权重文件模型训练

MT-UNet是一种结合Transformer和UNet优势的医学图像分割模型。该模型在Synapse和ACDC数据集上分别达到79.20%和91.61%的DSC评分。MT-UNet通过混合transformer结构实现多尺度特征融合，为医学图像分析提供新思路。项目开源代码和预训练权重，便于研究者复现结果和深入研究。

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

GithubImageNetPyTorchResNetffcv开源项目深度学习

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

AI2BMD - AI驱动的高精度蛋白质分子动力学模拟工具

AI2BMDGithub分子动力学模拟开源项目机器学习蛋白质计算化学

AI2BMD是一个结合人工智能和从头计算方法的蛋白质分子动力学模拟工具。它包含AIMD-Chig数据集、ViSNet和Geoformer模型架构，以及完整的模拟程序。该工具在保持高精度的同时提高了模拟效率，为蛋白质研究和药物开发提供了新的研究方法。

pytorch-lightning - 深度学习框架的全方位AI模型训练与部署解决方案

AI模型训练GithubLightning FabricPyTorch Lightning开源项目模型部署深度学习热门

深度学习框架Pytorch-Lightning 2.0版本现已推出，提供清晰稳定的API，支持AI模型的预训练、微调和部署。该框架轻松实现Pytorch代码组织，将科学研究与工程实现分离，帮助研究人员和工程师高效进行模型训练与部署。通过提供各种训练和部署选项以及兼容多种硬件和加速器，Pytorch-Lightning兼顾模型的灵活性和可扩展性，适应从初学者到专业AI研究的不同需求。

nncf - Neural Network Compression Framework：高效神经网络推理压缩算法

GithubNeural Network Compression FrameworkONNXOpenVINOPyTorchTensorFlow开源项目

Neural Network Compression Framework (NNCF) 提供一套后训练和训练时的优化算法，用于在 OpenVINO 中优化神经网络推理，保证最小的精度损失。NNCF 支持 PyTorch、TensorFlow 和 ONNX 等模型，并提供示例展示不同压缩算法的使用案例。NNCF 还支持自动化模型图转换、分布式训练和多种算法的无缝组合，支持将压缩后的 PyTorch 模型导出为 ONNX 检查点及将 TensorFlow 模型导出为 SavedModel 格式。

equiformer-pytorch - SE(3)/E(3)等变注意力网络的高效PyTorch实现

AIEquiformerGATv2GithubSE3 Transformers开源项目深度学习

Equiformer-pytorch是一个基于PyTorch的SE(3)/E(3)等变注意力网络实现。该项目采用MLP注意力机制和非线性消息传递，实现了最先进的性能。它支持可逆网络以提高内存效率，并集成了最新的球谐函数稀疏化技术，大幅提升计算效率。Equiformer-pytorch还提供边缘和邻接矩阵支持，适用于蛋白质折叠等各种3D原子图任务。

torchtitan - PyTorch原生大规模语言模型训练框架

GithubLLM训练Llama模型PyTorchtorchtitan分布式训练开源项目

torchtitan是基于PyTorch的大规模语言模型训练框架，展示了最新分布式训练功能。它采用简洁模块化设计，支持多种并行化技术，包括数据并行、张量并行和管道并行。框架还提供分布式检查点和Float8等先进特性，为Llama 3和Llama 2等模型预训练提供高效方案。torchtitan旨在展示PyTorch在大规模语言模型训练中的潜力。

med-seg-diff-pytorch - PyTorch实现的医学图像分割扩散模型

DDPMGithubPytorch医学图像分割开源项目扩散概率模型深度学习

med-seg-diff-pytorch是一个基于PyTorch的医学图像分割框架，采用扩散概率模型(DDPM)和特征级条件增强技术。该项目提供简易安装和使用方法，支持自定义数据集训练，并计划增加更多功能。它为医学图像分析领域提供了一个功能强大、使用灵活的开源工具。

MNN - 高效轻量的深度学习框架，支持多设备推理和训练

GithubMNN开源项目推理引擎深度学习框架轻量级高性能

MNN是一个高效轻量的深度学习框架，支持设备上的推理和训练。已被阿里巴巴30多个应用集成，覆盖直播、短视频、搜索推荐等70多种场景。MNN适用于嵌入式设备，支持TensorFlow、Caffe、ONNX等多种模型格式，并优化了ARM和x64 CPU及多种GPU的计算性能。通过MNN Workbench，用户可以下载预训练模型、进行可视化训练并一键部署到设备上。

相关项目

项目侧边栏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号