SchNetPack: A Deep Learning Toolbox For Atomistic Systems

Abstract

SchNetPack is a toolbox for the development and application of deep neural networks to the prediction of potential energy surfaces and other quantum-chemical properties of molecules and materials. It contains basic building blocks of atomistic neural networks, manages their training and provides simple access to common benchmark datasets. This allows for an easy implementation and evaluation of new models. For now, SchNetPack includes implementations of (weighted) atomcentered symmetry functions and the deep tensor neural network SchNet as well as ready-to-use scripts that allow to train these models on molecule and material datasets. Based upon the PyTorch deep learning framework, SchNetPack allows to efficiently apply the neural networks to large datasets with millions of reference calculations as well as parallelize the model across multiple GPUs. Finally, SchNetPack provides an interface to the Atomic Simulation Environment in order to make trained models easily accessible to researchers that are not yet familiar with neural networks.

References in corpus (25)

• 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
• Deep Potential Molecular Dynamics: a scalable model with the accuracy of quantum mechanics
• SchNet - a deep learning architecture for molecules and materials
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Comparing molecules and solids across structural and alchemical space
• Machine Learning Unifies the Modelling of Materials and Molecules
• By-passing the Kohn-Sham equations with machine learning
• Towards Exact Molecular Dynamics Simulations with Machine-Learned Force Fields
• Active learning of linearly parametrized interatomic potentials
• Machine learning prediction errors better than DFT accuracy
• Machine Learning of Molecular Electronic Properties in Chemical Compound Space
• Machine Learning Molecular Dynamics for the Simulation of Infrared Spectra
• How van der Waals interactions determine the unique properties of water
• How to represent crystal structures for machine learning: towards fast prediction of electronic properties
• Insightful classification of crystal structures using deep learning
• ANI-1: A data set of 20M off-equilibrium DFT calculations for organic molecules
• Alchemical and structural distribution based representation for improved QML
• Hierarchical modeling of molecular energies using a deep neural network
• WACSF - Weighted Atom-Centered Symmetry Functions as Descriptors in Machine Learning Potentials
• Unified Representation of Molecules and Crystals for Machine Learning
• Achieving DFT accuracy with a machine-learning interatomic potential: thermomechanics and defects in bcc ferromagnetic iron

Cited by in corpus (93)

• Machine Learning Force Fields
• Recent Advances and Applications of Deep Learning Methods in Materials Science
• DP-GEN: A concurrent learning platform for the generation of reliable deep learning based potential energy models
• A Performance and Cost Assessment of Machine Learning Interatomic Potentials
• Machine learning for molecular simulation
• Combining Machine Learning and Computational Chemistry for Predictive Insights Into Chemical Systems
• Atomistic Line Graph Neural Network for Improved Materials Property Predictions
• DeePMD-kit v2: A software package for Deep Potential models
• Opportunities and Challenges for Machine Learning in Materials Science
• Machine learning for electronically excited states of molecules
• SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effects
• FCHL revisited: faster and more accurate quantum machine learning
• Higher-Order Explanations of Graph Neural Networks via Relevant Walks
• sGDML: Constructing Accurate and Data Efficient Molecular Force Fields Using Machine Learning
• TorchMD: A deep learning framework for molecular simulations
• Combining SchNet and SHARC: The SchNarc machine learning approach for excited-state dynamics
• Deep Potentials for Materials Science
• Inverse design of 3d molecular structures with conditional generative neural networks
• Perspective on integrating machine learning into computational chemistry and materials science
• Coarse Graining Molecular Dynamics with Graph Neural Networks
• Quantum Machine Learning for Chemistry and Physics
• Machine Learning Coarse-Grained Potentials of Protein Thermodynamics
• Toward Explainable AI for Regression Models
• Molecular Force Fields with Gradient-Domain Machine Learning: Construction and Application to Dynamics of Small Molecules with Coupled Cluster Forces
• Gaussian Moments as Physically Inspired Molecular Descriptors for Accurate and Scalable Machine Learning Potentials
• OrbNet Denali: A machine learning potential for biological and organic chemistry with semi-empirical cost and DFT accuracy
• DeePMD-kit v3: A Multiple-Backend Framework for Machine Learning Potentials
• Strategies for the Construction of Machine-Learning Potentials for Accurate and Efficient Atomic-Scale Simulations
• SchNetPack 2.0: A neural network toolbox for atomistic machine learning
• High-throughput property-driven generative design of functional organic molecules
• Symmetry-adapted generation of 3d point sets for the targeted discovery of molecules
• PiNN: A Python Library for Building Atomic Neural Networks of Molecules and Materials
• Band gap prediction for large organic crystal structures with machine learning
• Deep Learning for UV Absorption Spectra with SchNarc: First Steps Towards Transferability in Chemical Compound Space
• Physically inspired deep learning of molecular excitations and photoemission spectra
• Accurate large-scale simulations of siliceous zeolites by neural network potentials
• Machine learning and excited-state molecular dynamics
• Graph Neural Networks Accelerated Molecular Dynamics
• Roaming leads to amino acid photodamage: A deep learning study of tyrosine
• Bayesian Force Fields from Active Learning for Simulation of Inter-Dimensional Transformation of Stanene
• A Perspective on Deep Learning for Molecular Modeling and Simulations
• Machine Learning Prediction of Accurate Atomization Energies of Organic Molecules from Low-Fidelity Quantum Chemical Calculations
• Molecular Force Fields with Gradient-Domain Machine Learning (GDML): Comparison and Synergies with Classical Force Fields
• Challenges for Machine Learning Force Fields in Reproducing Potential Energy Surfaces of Flexible Molecules
• Molecular Mechanics-Driven Graph Neural Network with Multiplex Graph for Molecular Structures
• BuRNN: Buffer Region Neural Network Approach for Polarizable-Embedding Neural Network / Molecular Mechanics Simulations
• Predicting tensorial molecular properties with equivariant machine-learning models
• A deep neural network for molecular wave functions in quasi-atomic minimal basis representation
• Relevance of Rotationally Equivariant Convolutions for Predicting Molecular Properties
• A reactive neural network framework for water-loaded acidic zeolites
• Infrared spectra at coupled cluster accuracy from neural network representations
• Towards Linearly Scaling and Chemically Accurate Global Machine Learning Force Fields for Large Molecules
• Fast and Sample-Efficient Interatomic Neural Network Potentials for Molecules and Materials Based on Gaussian Moments
• Learning from the Density to Correct Total Energy and Forces in First Principle Simulations
• Neural network with optimal neuron activation functions based on additive Gaussian process regression
• Heat flux for semi-local machine-learning potentials
• Modeling electronic response properties with an explicit-electron machine learning potential
• A Universal Framework for Accurate and Efficient Geometric Deep Learning of Molecular Systems
• A deep machine learning potential for atomistic simulation of Fe-Si-O systems under Earth's outer core conditions
• Kernel based quantum machine learning at record rate : Many-body distribution functionals as compact representations
• Reinforcement Learning for Molecular Design Guided by Quantum Mechanics
• chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics
• Degeneration of kernel regression with Matern kernels into low-order polynomial regression in high dimension
• Reducing Training Data Needs with Minimal Multilevel Machine Learning (M3L)
• Stress and heat flux via automatic differentiation
• Neural Networks Potential from the Bispectrum Component: A Case Study on Crystalline Silicon
• Analyzing Atomic Interactions in Molecules as Learned by Neural Networks
• A machine learning potential for simulating infrared spectra of nanosilicate clusters
• Mean-Field Density Matrix Decompositions
• Predicting hot-electron free energies from ground-state data
• Super-resolution in Molecular Dynamics Trajectory Reconstruction with Bi-Directional Neural Networks
• Construction of Machine Learned Force Fields with Quantum Chemical Accuracy: Applications and Chemical Insights
• Machine Learning for Screening Large Organic Molecules
• Towards Symbolic XAI -- Explanation Through Human Understandable Logical Relationships Between Features
• A Descriptor Is All You Need: Accurate Machine Learning of Nonadiabatic Coupling Vectors
• Machine learning at the atomic-scale
• Graph-neural-network predictions of solid-state NMR parameters from spherical tensor decomposition
• Machine learning frontier orbital energies of nanodiamonds
• SynCoTrain: A Dual Classifier PU-learning Framework for Synthesizability Prediction
• Developing Machine-Learned Potentials for Coarse-Grained Molecular Simulations: Challenges and Pitfalls
• Fortnet, a software package for training Behler-Parrinello neural networks
• Combining high-performance hardware, cloud computing, and deep learning frameworks to accelerate physical simulations: probing the Hopfield network
• Molecular relaxation by reverse diffusion with time step prediction
• Physical machine learning outperforms "human learning" in Quantum Chemistry
• Artificial Intelligence based Autonomous Molecular Design for Medical Therapeutic: A Perspective
• MLSolv-A: A Novel Machine Learning-Based Prediction of Solvation Free Energies from Pairwise Atomistic Interactions
• Anharmonic infrared spectra of cationic pyrene and superhydrogenated derivatives
• A Simple and Scalable Kernel Density Approach for Reliable Uncertainty Quantification in Atomistic Machine Learning
• Assessing zero-shot generalisation behaviour in graph-neural-network interatomic potentials
• HydroNet: Benchmark Tasks for Preserving Intermolecular Interactions and Structural Motifs in Predictive and Generative Models for Molecular Data
• REANN: A PyTorch-based End-to-End Multi-functional Deep Neural Network Package for Molecular, Reactive and Periodic Systems
• Analysis of Atomistic Representations Using Weighted Skip-Connections
• NQCDynamics.jl: A Julia Package for Nonadiabatic Quantum Classical Molecular Dynamics in the Condensed Phase