SchNetPack 2.0: A neural network toolbox for atomistic machine learning

Abstract

SchNetPack is a versatile neural networks toolbox that addresses both the requirements of method development and application of atomistic machine learning. Version 2.0 comes with an improved data pipeline, modules for equivariant neural networks as well as a PyTorch implementation of molecular dynamics. An optional integration with PyTorch Lightning and the Hydra configuration framework powers a flexible command-line interface. This makes SchNetPack 2.0 easily extendable with custom code and ready for complex training task such as generation of 3d molecular structures.

References in corpus (51)

• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• Escaping free-energy minima
• Neural Message Passing for Quantum Chemistry
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• SchNet - a deep learning architecture for molecules and materials
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Machine Learning Force Fields
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• PhysNet: A Neural Network for Predicting Energies, Forces, Dipole Moments and Partial Charges
• Machine learning for molecular simulation
• Combining Machine Learning and Computational Chemistry for Predictive Insights Into Chemical Systems
• By-passing the Kohn-Sham equations with machine learning
• Towards Exact Molecular Dynamics Simulations with Machine-Learned Force Fields
• Machine learning prediction errors better than DFT accuracy
• Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
• Machine Learning Molecular Dynamics for the Simulation of Infrared Spectra
• VAMPnets: Deep learning of molecular kinetics
• Deep neural network solution of the electronic Schrödinger equation
• Ab-Initio Solution of the Many-Electron Schrödinger Equation with Deep Neural Networks
• Machine learning a general purpose interatomic potential for silicon
• SchNet: A continuous-filter convolutional neural network for modeling quantum interactions
• Accurate sampling using Langevin dynamics
• SchNetPack: A Deep Learning Toolbox For Atomistic Systems
• Efficient stochastic thermostatting of path integral molecular dynamics
• i-PI 2.0: A Universal Force Engine for Advanced Molecular Simulations
• SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effects
• Alchemical and structural distribution based representation for improved QML
• Constrained Graph Variational Autoencoders for Molecule Design
• How to remove the spurious resonances from ring polymer molecular dynamics
• MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fields
• Equivariant message passing for the prediction of tensorial properties and molecular spectra
• Colored-noise thermostats à la carte
• 3D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Data
• 86 PFLOPS Deep Potential Molecular Dynamics simulation of 100 million atoms with ab initio accuracy
• TorchMD: A deep learning framework for molecular simulations
• Inverse design of 3d molecular structures with conditional generative neural networks
• Perspective on integrating machine learning into computational chemistry and materials science
• Unsupervised machine learning in atomistic simulations, between predictions and understanding
• Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physics
• Predicting molecular dipole moments by combining atomic partial charges and atomic dipoles
• Symmetry-adapted generation of 3d point sets for the targeted discovery of molecules
• Autonomous robotic nanofabrication with reinforcement learning
• Band gap prediction for large organic crystal structures with machine learning
• Physically inspired deep learning of molecular excitations and photoemission spectra
• Machine learning and excited-state molecular dynamics
• Equivariant Flows: sampling configurations for multi-body systems with symmetric energies
• SE(3)-equivariant prediction of molecular wavefunctions and electronic densities
• Accurate Machine Learned Quantum-Mechanical Force Fields for Biomolecular Simulations
• Symmetry-Aware Actor-Critic for 3D Molecular Design

Cited by in corpus (21)

• DeePMD-kit v2: A software package for Deep Potential models
• DeePMD-kit v3: A Multiple-Backend Framework for Machine Learning Potentials
• MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflows
• The Evolution of Machine Learning Potentials for Molecules, Reactions and Materials
• TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulations
• Generative diffusion model for surface structure discovery
• Neural Network Potential with Multi-Resolution Approach Enables Accurate Prediction of Reaction Free Energies in Solution
• chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics
• MGNN: Moment Graph Neural Network for Universal Molecular Potentials
• Stress and heat flux via automatic differentiation
• Analyzing Atomic Interactions in Molecules as Learned by Neural Networks
• FeNNol: an Efficient and Flexible Library for Building Force-field-enhanced Neural Network Potentials
• Metatensor and metatomic: foundational libraries for interoperable atomistic machine learning
• Towards Symbolic XAI -- Explanation Through Human Understandable Logical Relationships Between Features
• A Descriptor Is All You Need: Accurate Machine Learning of Nonadiabatic Coupling Vectors
• Reconstructions and Dynamics of $\beta$-Lithium Thiophosphate Surfaces
• EOSnet: Embedded Overlap Structures for Graph Neural Networks in Predicting Material Properties
• Molecular relaxation by reverse diffusion with time step prediction
• Guest Editorial: Special Topic on Software for Atomistic Machine Learning
• Anharmonic infrared spectra of cationic pyrene and superhydrogenated derivatives
• Faster Molecular Dynamics with Neural Network Potentials via Distilled Multiple Time-Stepping and Non-Conservative Forces