SchNetPack 2.0: A neural network toolbox for atomistic machine learning
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 29 papers in the map (data through 14 Oct 2026). Builds on 36 of its 87 references that are in the map. First posted on arXiv 11 Dec 2022.
Builds on
36 of 87 references.
References · 36
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 746
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 708
- 2016TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systemscited by 664
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
- 2020Directional Message Passing for Molecular Graphscited by 561
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 474
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 441
- 20183D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Datacited by 354
- 2022Learning Local Equivariant Representations for Large-Scale Atomistic Dynamicscited by 256
- 2020Machine Learning Force Fieldscited by 207
24 more
- 2019Ab-Initio Solution of the Many-Electron Schr"odinger Equation with Deep Neural Networkscited by 200
- 2018Constrained Graph Variational Autoencoders for Molecule Designcited by 185
- 2019Symmetry-adapted generation of 3d point sets for the targeted discovery of moleculescited by 172
- 2017VAMPnets: Deep learning of molecular kineticscited by 136
- 2019Deep neural network solution of the electronic Schr"odinger equationcited by 127
- 2019Machine learning for molecular simulationcited by 110
- 2021SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effectscited by 104
- 2021Inverse design of 3d molecular structures with conditional generative neural networkscited by 93
- 2019Unifying machine learning and quantum chemistry -- a deep neural network for molecular wavefunctionscited by 73
- 2016By-passing the Kohn-Sham equations with machine learningcited by 67
- 2021SE(3)-equivariant prediction of molecular wavefunctions and electronic densitiescited by 67
- 2019Equivariant Flows: sampling configurations for multi-body systems with symmetric energiescited by 57
- 2020Symmetry-Aware Actor-Critic for 3D Molecular Designcited by 55
- 2020TorchMD: A deep learning framework for molecular simulationscited by 55
- 2020Kohn-Sham equations as regularizer: building prior knowledge into machine-learned physicscited by 36
- 2017Machine Learning Molecular Dynamics for the Simulation of Infrared Spectracited by 30
- 2020Predicting molecular dipole moments by combining atomic partial charges and atomic dipolescited by 18
- 202086 PFLOPS Deep Potential Molecular Dynamics simulation of 100 million atoms with ab initio accuracycited by 15
- 2020Machine learning of solvent effects on molecular spectra and reactionscited by 14
- 2022Accurate Machine Learned Quantum-Mechanical Force Fields for Biomolecular Simulationscited by 9
- 2020Autonomous robotic nanofabrication with reinforcement learningcited by 8
- 2018Band gap prediction for large organic crystal structures with machine learningcited by 5
- 2020Machine learning and excited-state molecular dynamicscited by 2
- 2022Automatic Identification of Chemical Moietiescited by 2
Built on by
Citing papers per year: 2023: 9; 2024: 8; 2025: 6; 2026: 6.
29 papers.
Sort:
By area:
- 2023TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentialscited by 54
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 41
- 2023Learning invariant representations of time-homogeneous stochastic dynamical systemscited by 17
- 2025High-performance training and inference for deep equivariant interatomic potentialscited by 17
- 2024TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulationscited by 15
- 2024Neural Conditional Probability for Uncertainty Quantificationcited by 14
- 2024chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physicscited by 12
- 2023Data efficiency and extrapolation trends in neural network interatomic potentialscited by 10
- 2023MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflowscited by 10
- 2024Molecular relaxation by reverse diffusion with time step predictioncited by 9
- 2024Towards Symbolic XAI -- Explanation Through Human Understandable Logical Relationships Between Featurescited by 7
- 2024FeNNol: an Efficient and Flexible Library for Building Force-field-enhanced Neural Network Potentialscited by 6
17 more
- 2025OBELiX: A Curated Dataset of Crystal Structures and Experimentally Measured Ionic Conductivities for Lithium Solid-State Electrolytescited by 5
- 2025Efficient Parametric SVD of Koopman Operator for Stochastic Dynamical Systemscited by 5
- 2025HollowFlow: Efficient Sample Likelihood Evaluation using Hollow Message Passingcited by 5
- 2023Gradual Optimization Learning for Conformational Energy Minimizationcited by 4
- 2025CLOUD: A Scalable and Physics-Informed Foundation Model for Crystal Representation Learningcited by 3
- 2023Stress and heat flux via automatic differentiationcited by 1
- 2023Multiscale Hodge Scattering Networks for Data Analysiscited by 1
- 2026Faster Molecular Dynamics with Neural Network Potentials via Distilled Multiple Time-Stepping and Non-Conservative Forcescited by 1
- 2026Generative Pseudo-Force Fields for Molecular Generationcited by 1
- 2026Implicit Machine Learning Force Fields Accelerate Molecular Dynamics Simulationscited by 1
- 2023Chemellia: An Ecosystem for Atomistic Scientific Machine Learningcited by 0
- 2024Infrared Spectra Prediction for Diazo Groups Utilizing a Machine Learning Approach with Structural Attention Mechanismcited by 0
- 2024it Asparagus: A Toolkit for Autonomous, User-Guided Construction of Machine-Learned Potential Energy Surfacescited by 0
- 2025A Descriptor Is All You Need: Accurate Machine Learning of Nonadiabatic Coupling Vectorscited by 0
- 2026GEqTrain: A Configuration-Driven Framework for Retargeting Equivariant Graph Neural Networks Across 3D Scientific Taskscited by 0
- 2026Temperature-Driven Sequential Modeling for the Prediction of Annual Power Conversion Efficiency Profiles of Organic Photovoltaic Materials: Douala Case Studycited by 0
- 2026The Mechanics of Delta Learning: Target Design for Generalizable Scientific Machine Learningcited by 0
- 2026The Mechanics of Delta Learning: Target Design for Generalizable Scientific Machine Learning23 Sep 2026
- 2026Temperature-Driven Sequential Modeling for the Prediction of Annual Power Conversion Efficiency Profiles of Organic Photovoltaic Materials: Douala Case Study9 Aug 2026
- 2026Implicit Machine Learning Force Fields Accelerate Molecular Dynamics Simulations31 Jul 2026
- 2026GEqTrain: A Configuration-Driven Framework for Retargeting Equivariant Graph Neural Networks Across 3D Scientific Tasks21 Jul 2026
- 2026Generative Pseudo-Force Fields for Molecular Generation18 May 2026
- 2026Faster Molecular Dynamics with Neural Network Potentials via Distilled Multiple Time-Stepping and Non-Conservative Forces16 Feb 2026
- 2025HollowFlow: Efficient Sample Likelihood Evaluation using Hollow Message Passing24 Oct 2025
- 2025Efficient Parametric SVD of Koopman Operator for Stochastic Dynamical Systems9 Jul 2025
- 2025CLOUD: A Scalable and Physics-Informed Foundation Model for Crystal Representation Learning19 Jun 2025
- 2025A Descriptor Is All You Need: Accurate Machine Learning of Nonadiabatic Coupling Vectors29 May 2025
- 2025High-performance training and inference for deep equivariant interatomic potentials22 Apr 2025
- 2025OBELiX: A Curated Dataset of Crystal Structures and Experimentally Measured Ionic Conductivities for Lithium Solid-State Electrolytes20 Feb 2025
17 more
- 2024Towards Symbolic XAI -- Explanation Through Human Understandable Logical Relationships Between Features30 Aug 2024
- 2024chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics28 Aug 2024
- 2024it Asparagus: A Toolkit for Autonomous, User-Guided Construction of Machine-Learned Potential Energy Surfaces21 Jul 2024
- 2024Neural Conditional Probability for Uncertainty Quantification1 Jul 2024
- 2024FeNNol: an Efficient and Flexible Library for Building Force-field-enhanced Neural Network Potentials2 May 2024
- 2024Molecular relaxation by reverse diffusion with time step prediction16 Apr 2024
- 2024TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulations27 Feb 2024
- 2024Infrared Spectra Prediction for Diazo Groups Utilizing a Machine Learning Approach with Structural Attention Mechanism5 Feb 2024
- 2023Multiscale Hodge Scattering Networks for Data Analysis17 Nov 2023
- 2023Gradual Optimization Learning for Conformational Energy Minimization5 Nov 2023
- 2023MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflows31 Oct 2023
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space4 Oct 2023
- 2023Learning invariant representations of time-homogeneous stochastic dynamical systems19 Jul 2023
- 2023TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials10 Jun 2023
- 2023Chemellia: An Ecosystem for Atomistic Scientific Machine Learning19 May 2023
- 2023Stress and heat flux via automatic differentiation2 May 2023
- 2023Data efficiency and extrapolation trends in neural network interatomic potentials12 Feb 2023
- 2026Generative Pseudo-Force Fields for Molecular Generationcited by 1
- 2026Generative Pseudo-Force Fields for Molecular Generation18 May 2026
- 2023TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentialscited by 54
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 41
- 2023Learning invariant representations of time-homogeneous stochastic dynamical systemscited by 17
- 2025High-performance training and inference for deep equivariant interatomic potentialscited by 17
- 2024TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulationscited by 15
- 2024Neural Conditional Probability for Uncertainty Quantificationcited by 14
- 2024chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physicscited by 12
- 2023Data efficiency and extrapolation trends in neural network interatomic potentialscited by 10
- 2023MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflowscited by 10
- 2024Molecular relaxation by reverse diffusion with time step predictioncited by 9
- 2024Towards Symbolic XAI -- Explanation Through Human Understandable Logical Relationships Between Featurescited by 7
- 2024FeNNol: an Efficient and Flexible Library for Building Force-field-enhanced Neural Network Potentialscited by 6
12 most cited of 28 All 28 (JSON)
- 2026The Mechanics of Delta Learning: Target Design for Generalizable Scientific Machine Learning23 Sep 2026
- 2026Temperature-Driven Sequential Modeling for the Prediction of Annual Power Conversion Efficiency Profiles of Organic Photovoltaic Materials: Douala Case Study9 Aug 2026
- 2026Implicit Machine Learning Force Fields Accelerate Molecular Dynamics Simulations31 Jul 2026
- 2026GEqTrain: A Configuration-Driven Framework for Retargeting Equivariant Graph Neural Networks Across 3D Scientific Tasks21 Jul 2026
- 2026Faster Molecular Dynamics with Neural Network Potentials via Distilled Multiple Time-Stepping and Non-Conservative Forces16 Feb 2026
- 2025HollowFlow: Efficient Sample Likelihood Evaluation using Hollow Message Passing24 Oct 2025
- 2025Efficient Parametric SVD of Koopman Operator for Stochastic Dynamical Systems9 Jul 2025
- 2025CLOUD: A Scalable and Physics-Informed Foundation Model for Crystal Representation Learning19 Jun 2025
- 2025A Descriptor Is All You Need: Accurate Machine Learning of Nonadiabatic Coupling Vectors29 May 2025
- 2025High-performance training and inference for deep equivariant interatomic potentials22 Apr 2025
- 2025OBELiX: A Curated Dataset of Crystal Structures and Experimentally Measured Ionic Conductivities for Lithium Solid-State Electrolytes20 Feb 2025
- 2024Towards Symbolic XAI -- Explanation Through Human Understandable Logical Relationships Between Features30 Aug 2024
12 newest of 28 All 28 (JSON)