SchNetPack 2.0：一个用于原子级机器学习的神经网络工具箱
化学物理
2023-04-26 v1 机器学习
摘要
SchNetPack 是一个通用的神经网络工具箱，满足了原子级机器学习的方法开发与应用两方面的需求。2.0 版本带来了改进的数据流水线、等变神经网络模块以及分子动力学的 PyTorch 实现。与 PyTorch Lightning 和 Hydra 配置框架的可选集成提供了一个灵活的命令行接口。这使得 SchNetPack 2.0 易于通过自定义代码扩展，并可用于如生成三维分子结构等复杂训练任务。
引用
@article{arxiv.2212.05517,
title = {SchNetPack 2.0: A neural network toolbox for atomistic machine learning},
author = {Kristof T. Schütt and Stefaan S. P. Hessmann and Niklas W. A. Gebauer and Jonas Lederer and Michael Gastegger},
journal= {arXiv preprint arXiv:2212.05517},
year = {2023}
}