mlcolvar

Machine learning collective variables for enhanced sampling

Website: https://mlcolvar.readthedocs.io

Github: luigibonati/mlcolvar

Publication: A unified framework for machine learning collective variables for enhanced sampling simulations: mlcolvar (Bonati et al., 2023)


mlcolvar is a Python library aimed to help design data-driven collective-variables (CVs) for enhanced sampling simulations. The key features are:

  1. A unified framework to help test and use (some) of the CVs proposed in the literature.
  2. A modular interface to simplify the development of new approaches and the contamination between them.
  3. A streamlined distribution of CVs in the context of advanced sampling.

The library is built upon the PyTorch ML library as well as the Lightning high-level framework.


Some of the CVs which are implemented, organized by learning setting:

  • Unsupervised: PCA, (Variational) AutoEncoders [1,2]
  • Supervised: LDA [3], DeepLDA [4], DeepTDA [5]
  • Time-informed: TICA [6], DeepTICA/SRVs [7,8], VDE [9]

And many others can be implemented based on the building blocks or with simple modifications. Check out the tutorials and the examples section of the documentation.

References

2023

  1. JCP
    2023-jcp-mlcolvar.png
    A unified framework for machine learning collective variables for enhanced sampling simulations: mlcolvar
    Luigi Bonati, Enrico Trizio, Andrea Rizzi, and Michele Parrinello
    The Journal of Chemical Physics, Jul 2023