SchNetPack: A Deep Learning Toolbox For Atomistic Systems.

By Kristof T. Schütt, Pan Kessel, Michael Gastegger, K. A. Nicoli, Alexandre Tkatchenko, K. Müller. 2018

389 citations recorded by the source corpus.

SchNetPack is a toolbox for the development and application of deep neural networks that predict 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) atom-centered symmetry functions and the deep tensor neural network SchNet, as well as ready-to-use scripts that allow one to train these models on molecule and material datasets. Based on the PyTorch deep learning framework, SchNetPack allows one 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.

Subfields: Computational Theory and Mathematics, Materials Chemistry

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computational Theory and Mathematics

  1. Tess E. Smidt
  2. Frank No'e
  3. Mario Geiger
  4. Michael Gastegger
  5. Grgoire Montavon
  6. Kristof T. Schütt
  7. Klaus‐Robert Müller 
  8. Florian Hase

In Materials Chemistry

  1. Frank No'e
  2. Tess E. Smidt
  3. Michael Gastegger
  4. Mario Geiger
  5. Klaus‐Robert Müller 
  6. Cecilia Clementi
  7. Kristof T. Schütt
  8. Grgoire Montavon

Citations in the corpus