Skip to content
@lab-cosmo

Laboratory of Computational Science and Modeling

Hi there! This is the github page of the COSMO lab. In this repository you can find the software developed by our group, collaborators and friends.

We publish our code because we believe in open science and want the community to both benefit from our work and contribute to it.

Clone, enjoy... and cite

External Projects

Besides the repositories we pin below we develop the following projects.

metatrain: a command line tool to train atomistic machine learning model for any kind of target

metatomic: a standard interface to load and execute ML models in various simulation engines

metatensor: a data storage and exchange format created for atomistic machine learning

i-pi a universal force engine designed to be used together with an ab-initio, empirical or machine-learning force field (or a combination of these!) to run molecular dynamics simulations, including nuclear quantum effects.

scikit-matter: a collection of scikit-learn compatible utilities that implement methods born out of the materials science and chemistry communities.

featomic: a library to compute representations for atomistic machine learning

Tutorials and Examples:

atomistic-cookbook an online, interactive collection of "recipes" to perform different kinds of atomistic simulations, from path integral dynamics to the training of a ML model

kernel-tutorials a set of utilities and pedagogic notebooks for the use of linear and kernel methods in atomistic modeling

sampling-tutorial a set of exercise files and handouts for a sampling tutorial, from thermostatting to non-linear dimensionality reduction

Legacy code:

We list below a collection of legacy codes and tools developed in our lab. They are functional and working, but no longer maintained.

gle4md colored-noise thermostats for molecular dynamics: http://gle4md.org/

pamm tools for the Probabilistic Analysis of Molecular Motifs method

sketchmap a suite of programs to perform non-linear dimensionality reduction: http://sketchmap.org

Pinned Loading

  1. atomistic-cookbook atomistic-cookbook Public

    A collection of simulation recipes for the atomic-scale modeling of materials and molecules

    Python 55 12

  2. chemiscope chemiscope Public

    An interactive structure/property explorer for materials and molecules

    TypeScript 183 44

  3. torch-pme torch-pme Public

    Particle-mesh based calculations of long-range interactions in PyTorch

    Python 81 12

  4. upet upet Public

    Universal machine-learning models for advanced atomistic simulations

    Python 232 25

  5. flashmd flashmd Public

    A universal ML model to predict molecular dynamics trajectories with long time steps

    Python 57 8

  6. shiftml shiftml Public

    A python package for the prediction of chemical shieldings of organic solids and beyond.

    Python 22 4

Repositories

Showing 10 of 53 repositories
  • lorem-jax Public

    Learning Long-Range Representations with Equivariant Messages

    lab-cosmo/lorem-jax's past year of commit activity
    Python 9 BSD-3-Clause 4 0 3 Updated Sep 9, 2026
  • flashmd Public

    A universal ML model to predict molecular dynamics trajectories with long time steps

    lab-cosmo/flashmd's past year of commit activity
    Python 57 Apache-2.0 8 0 7 Updated Sep 9, 2026
  • upet Public

    Universal machine-learning models for advanced atomistic simulations

    lab-cosmo/upet's past year of commit activity
    Python 232 BSD-3-Clause 25 2 2 Updated Sep 9, 2026
  • atomistic-cookbook Public

    A collection of simulation recipes for the atomic-scale modeling of materials and molecules

    lab-cosmo/atomistic-cookbook's past year of commit activity
    Python 55 BSD-3-Clause 12 11 11 Updated Sep 8, 2026
  • sphericart Public

    Multi-language library for the calculation of spherical harmonics in Cartesian coordinates

    lab-cosmo/sphericart's past year of commit activity
    C++ 103 MIT 17 9 1 Updated Sep 7, 2026
  • chemiscope Public

    An interactive structure/property explorer for materials and molecules

    lab-cosmo/chemiscope's past year of commit activity
    TypeScript 183 BSD-3-Clause 44 15 (1 issue needs help) 1 Updated Sep 1, 2026
  • lab-cosmo/wigner_kernels's past year of commit activity
    Python 2 1 0 0 Updated Aug 28, 2026
  • shiftml Public

    A python package for the prediction of chemical shieldings of organic solids and beyond.

    lab-cosmo/shiftml's past year of commit activity
    Python 22 BSD-3-Clause 4 2 (1 issue needs help) 0 Updated Aug 24, 2026
  • pet-jax Public

    Universal PET models in JAX

    lab-cosmo/pet-jax's past year of commit activity
    Python 8 MIT 1 0 2 Updated Aug 17, 2026
  • jax-pme Public

    Particle-mesh based calculations of long-range interactions in JAX

    lab-cosmo/jax-pme's past year of commit activity
    Python 30 5 0 0 Updated Aug 9, 2026