Software Resources

Software Resources

Materials Science

For ML packages specific to materials science, refer to the following list. Most of the examples below are pulled directly from this resource.

ML Frameworks

The following are useful frameworks for training machine learning models:
  • Star  scikit-learn : The go-to standard for "conventional" (i.e. not graph-related) classification, regression, and clustering machine learning algorithms. This is the package you should start with when getting familiar with machine learning.
  • Star  PyTorch : A Python library for training and running neural networks. This is the deep learning framework we use in the group.
  •  PyTorch Geometric : PyTorch-based library for graph neural networks.
  •  PyTorch Lightning : Helps to remove boilerplate in PyTorch codes
  •  PySR : Symbolic regression, for when having an equation is desirable.
  •  Optuna : Hyperparameter tuning

ML Interatomic Potentials

To find the top pre-trained ML interatomic potentials for materials science, refer to  Matbench Discovery. 
For some pre-made calculation tasks for ML interatomic potentials, check out  matcalc .
To train your own ML interatomic potentials, here are some general-purpose libraries:
  •  Graph PES 
  •  Autoplex