JuDFTteam / best-of-atomistic-machine-learning

πŸ† A ranked list of awesome atomistic machine learning projects βš›οΈπŸ§¬πŸ’Ž.
Creative Commons Attribution Share Alike 4.0 International
272 stars 19 forks source link
ai4science atomistic-machine-learning awesome-list best-of-list computational-chemistry computational-materials-science condensed-matter density-functional-theory drug-discovery electronic-structure interatomic-potentials materials-discovery materials-informatics molecular-dynamics quantum-chemistry scientific-computing scientific-machine-learning surrogate-models

Best of Atomistic Machine Learning βš›οΈπŸ§¬πŸ’Ž

πŸ†  A ranked list of awesome atomistic machine learning (AML) projects. Updated quarterly.

DOI

This curated list contains 360 awesome open-source projects with a total of 180K stars grouped into 22 categories. All projects are ranked by a project-quality score, which is calculated based on various metrics automatically collected from GitHub and different package managers. If you like to add or update projects, feel free to open an issue, submit a pull request, or directly edit the projects.yaml.

The current focus of this list is more on simulation data rather than experimental data, and more on materials rather than drug design. Nevertheless, contributions from other fields are warmly welcome!

πŸ§™β€β™‚οΈ Discover other best-of lists or create your own.

Contents

Explanation


Active learning

Back to top

Projects that focus on enabling active learning, iterative learning schemes for atomistic ML.

FLARE (πŸ₯‡20 Β· ⭐ 270) - An open-source Python package for creating fast and accurate interatomic potentials. MIT C++ ML-IAP - [GitHub](https://github.com/mir-group/flare) (πŸ‘¨β€πŸ’» 36 Β· πŸ”€ 63 Β· πŸ“₯ 7 Β· πŸ“¦ 11 Β· πŸ“‹ 200 - 15% open Β· ⏱️ 11.06.2024): ``` git clone https://github.com/mir-group/flare ```
IPSuite (πŸ₯ˆ16 Β· ⭐ 16) - A Python toolkit for FAIR development and deployment of machine-learned interatomic potentials. EPL-2.0 ML-IAP MD workflows HTC - [GitHub](https://github.com/zincware/IPSuite) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 7 Β· πŸ“¦ 2 Β· πŸ“‹ 120 - 50% open Β· ⏱️ 26.06.2024): ``` git clone https://github.com/zincware/IPSuite ``` - [PyPi](https://pypi.org/project/ipsuite) (πŸ“₯ 44 / month): ``` pip install ipsuite ```
Finetuna (πŸ₯‰10 Β· ⭐ 42) - Active Learning for Machine Learning Potentials. MIT - [GitHub](https://github.com/ulissigroup/finetuna) (πŸ‘¨β€πŸ’» 11 Β· πŸ”€ 11 Β· πŸ“‹ 20 - 25% open Β· ⏱️ 15.05.2024): ``` git clone https://github.com/ulissigroup/finetuna ```
ACEHAL (πŸ₯‰5 Β· ⭐ 11 Β· πŸ’€) - Hyperactive Learning (HAL) Python interface for building Atomic Cluster Expansion potentials. Unlicensed Julia - [GitHub](https://github.com/ACEsuit/ACEHAL) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 6 Β· πŸ“‹ 10 - 40% open Β· ⏱️ 21.09.2023): ``` git clone https://github.com/ACEsuit/ACEHAL ```
Show 1 hidden projects... - flare++ (πŸ₯‰10 Β· ⭐ 35 Β· πŸ’€) - A many-body extension of the FLARE code. MIT C++ ML-IAP


Biomolecules

Back to top

Projects that focus on biomolecules, protein structure, protein folding, etc. using atomistic ML.

AlphaFold (πŸ₯‡23 Β· ⭐ 12K) - Open source code for AlphaFold. Apache-2 - [GitHub](https://github.com/google-deepmind/alphafold) (πŸ‘¨β€πŸ’» 20 Β· πŸ”€ 2.1K Β· πŸ“¦ 14 Β· πŸ“‹ 840 - 28% open Β· ⏱️ 08.05.2024): ``` git clone https://github.com/deepmind/alphafold ```
Uni-Fold (πŸ₯‰16 Β· ⭐ 360) - An open-source platform for developing protein models beyond AlphaFold. Apache-2 - [GitHub](https://github.com/dptech-corp/Uni-Fold) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 64 Β· πŸ“₯ 3.5K Β· πŸ“‹ 70 - 27% open Β· ⏱️ 08.01.2024): ``` git clone https://github.com/dptech-corp/Uni-Fold ```


Community resources

Back to top

Projects that collect atomistic ML resources or foster communication within community.

πŸ”— AI for Science Map - Interactive mindmap of the AI4Science research field, including atomistic machine learning, including papers,..

πŸ”— Atomic Cluster Expansion - Atomic Cluster Expansion (ACE) community homepage.

πŸ”— CrystaLLM - Generate a crystal structure from a composition. language-models generative pre-trained transformer

πŸ”— matsci.org - A community forum for the discussion of anything materials science, with a focus on computational materials science..

πŸ”— Matter Modeling Stack Exchange - Machine Learning - Forum StackExchange, site Matter Modeling, ML-tagged questions.

Best-of Machine Learning with Python (πŸ₯‡22 Β· ⭐ 16K) - A ranked list of awesome machine learning Python libraries. Updated weekly. CC-BY-4.0 general-ml Python - [GitHub](https://github.com/ml-tooling/best-of-ml-python) (πŸ‘¨β€πŸ’» 45 Β· πŸ”€ 2.2K Β· πŸ“‹ 53 - 35% open Β· ⏱️ 07.06.2024): ``` git clone https://github.com/ml-tooling/best-of-ml-python ```
MatBench (πŸ₯‡18 Β· ⭐ 100) - Matbench: Benchmarks for materials science property prediction. MIT datasets benchmarking - [GitHub](https://github.com/materialsproject/matbench) (πŸ‘¨β€πŸ’» 25 Β· πŸ”€ 41 Β· πŸ“¦ 15 Β· πŸ“‹ 57 - 54% open Β· ⏱️ 20.01.2024): ``` git clone https://github.com/materialsproject/matbench ``` - [PyPi](https://pypi.org/project/matbench) (πŸ“₯ 760 / month): ``` pip install matbench ```
GT4SD - Generative Toolkit for Scientific Discovery (πŸ₯ˆ17 Β· ⭐ 320) - Gradio apps of generative models in GT4SD. MIT generative pre-trained drug-discovery - [GitHub](https://github.com/GT4SD/gt4sd-core) (πŸ‘¨β€πŸ’» 20 Β· πŸ”€ 66 Β· πŸ“‹ 99 - 1% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/GT4SD/gt4sd-core ```
Graph-based Deep Learning Literature (πŸ₯ˆ16 Β· ⭐ 4.7K) - links to conference publications in graph-based deep learning. MIT general-ml rep-learn - [GitHub](https://github.com/naganandy/graph-based-deep-learning-literature) (πŸ‘¨β€πŸ’» 12 Β· πŸ”€ 740 Β· ⏱️ 30.03.2024): ``` git clone https://github.com/naganandy/graph-based-deep-learning-literature ```
MatBench Discovery (πŸ₯ˆ16 Β· ⭐ 77) - An evaluation framework for machine learning models simulating high-throughput materials discovery. MIT datasets benchmarking - [GitHub](https://github.com/janosh/matbench-discovery) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 9 Β· πŸ“¦ 2 Β· πŸ“‹ 33 - 12% open Β· ⏱️ 24.06.2024): ``` git clone https://github.com/janosh/matbench-discovery ``` - [PyPi](https://pypi.org/project/matbench-discovery) (πŸ“₯ 62 / month): ``` pip install matbench-discovery ```
AI for Science Resources (πŸ₯ˆ12 Β· ⭐ 440) - List of resources for AI4Science research, including learning resources. GPL-3.0 license - [GitHub](https://github.com/divelab/AIRS) (πŸ‘¨β€πŸ’» 28 Β· πŸ”€ 55 Β· πŸ“‹ 13 - 7% open Β· ⏱️ 14.06.2024): ``` git clone https://github.com/divelab/AIRS ```
GNoME Explorer (πŸ₯‰10 Β· ⭐ 840 Β· πŸ’€) - Graph Networks for Materials Exploration Database. Apache-2 datasets materials-discovery - [GitHub](https://github.com/google-deepmind/materials_discovery) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 130 Β· πŸ“‹ 20 - 80% open Β· ⏱️ 02.12.2023): ``` git clone https://github.com/google-deepmind/materials_discovery ```
Awesome Materials Informatics (πŸ₯‰9 Β· ⭐ 350) - Curated list of known efforts in materials informatics, i.e. in modern materials science. Custom - [GitHub](https://github.com/tilde-lab/awesome-materials-informatics) (πŸ‘¨β€πŸ’» 19 Β· πŸ”€ 80 Β· ⏱️ 22.06.2024): ``` git clone https://github.com/tilde-lab/awesome-materials-informatics ```
MoLFormers UI (πŸ₯‰9 Β· ⭐ 220 Β· πŸ’€) - A family of foundation models trained on chemicals. Apache-2 transformer language-models pre-trained drug-discovery - [GitHub](https://github.com/IBM/molformer) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 39 Β· πŸ“‹ 18 - 44% open Β· ⏱️ 16.10.2023): ``` git clone https://github.com/IBM/molformer ```
optimade.science (πŸ₯‰9 Β· ⭐ 8) - A sky-scanner Optimade browser-only GUI. MIT datasets - [GitHub](https://github.com/tilde-lab/optimade.science) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 2 Β· πŸ“‹ 25 - 28% open Β· ⏱️ 10.06.2024): ``` git clone https://github.com/tilde-lab/optimade.science ```
Awesome Neural Geometry (πŸ₯‰8 Β· ⭐ 870) - A curated collection of resources and research related to the geometry of representations in the brain, deep networks,.. Unlicensed educational rep-learn - [GitHub](https://github.com/neurreps/awesome-neural-geometry) (πŸ‘¨β€πŸ’» 11 Β· πŸ”€ 53 Β· ⏱️ 14.02.2024): ``` git clone https://github.com/neurreps/awesome-neural-geometry ```
The Collection of Database and Dataset Resources in Materials Science (πŸ₯‰6 Β· ⭐ 240) - A list of databases, datasets and books/handbooks where you can find materials properties for machine learning.. Unlicensed datasets - [GitHub](https://github.com/sedaoturak/data-resources-for-materials-science) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 40 Β· πŸ“‹ 2 - 50% open Β· ⏱️ 07.06.2024): ``` git clone https://github.com/sedaoturak/data-resources-for-materials-science ```
Show 4 hidden projects... - A Highly Opinionated List of Open-Source Materials Informatics Resources (πŸ₯‰8 Β· ⭐ 110 Β· πŸ’€) - A Highly Opinionated List of Open Source Materials Informatics Resources. MIT - Does this material exist? (πŸ₯‰3 Β· ⭐ 15) - Vote on whether you think predicted crystal structures could be synthesised. MIT for-fun materials-discovery - MateriApps (πŸ₯‰1) - A Portal Site of Materials Science Simulation. Unlicensed - GitHub topic materials-informatics - Unlicensed


Datasets

Back to top

Datasets, databases and trained models for atomistic ML.

πŸ”— Catalysis Hub - A web-platform for sharing data and software for computational catalysis research!.

πŸ”— Citrination Datasets - AI-Powered Materials Data Platform. Open Citrination has been decommissioned.

πŸ”— crystals.ai - Curated datasets for reproducible AI in materials science.

πŸ”— DeepChem Models - DeepChem models on HuggingFace. pre-trained language-models

πŸ”— JARVIS-Leaderboard ( ⭐ 53) - Explore State-of-the-Art Materials Design Methods: https://www.nature.com/articles/s41524-024-01259-w. benchmarking

πŸ”— Materials Project - Charge Densities - Materials Project has started offering charge density information available for download via their public API.

πŸ”— matterverse.ai - Database of yet-to-be-sythesized materials predicted using state-of-the-art machine learning algorithms.

πŸ”— NRELMatDB - Computational materials database with the specific focus on materials for renewable energy applications including, but..

πŸ”— Quantum-Machine.org Datasets - Collection of datasets, including QM7, QM9, etc. MD, DFT. Small organic molecules, mostly.

πŸ”— sGDML Datasets - MD17, MD22, DFT datasets.

πŸ”— MoleculeNet - A Benchmark for Molecular Machine Learning. benchmarking

πŸ”— ZINC15 - A free database of commercially-available compounds for virtual screening. ZINC contains over 230 million purchasable.. graph biomolecules

πŸ”— ZINC20 - A free database of commercially-available compounds for virtual screening. ZINC contains over 230 million purchasable.. graph biomolecules

OPTIMADE Python tools (πŸ₯‡24 Β· ⭐ 62) - Tools for implementing and consuming OPTIMADE APIs in Python. MIT - [GitHub](https://github.com/Materials-Consortia/optimade-python-tools) (πŸ‘¨β€πŸ’» 28 Β· πŸ”€ 41 Β· πŸ“¦ 41 Β· πŸ“‹ 430 - 20% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/Materials-Consortia/optimade-python-tools ``` - [PyPi](https://pypi.org/project/optimade) (πŸ“₯ 5.9K / month): ``` pip install optimade ``` - [Conda](https://anaconda.org/conda-forge/optimade) (πŸ“₯ 79K Β· ⏱️ 11.06.2024): ``` conda install -c conda-forge optimade ```
MPContribs (πŸ₯‡23 Β· ⭐ 34) - Platform for materials scientists to contribute and disseminate their materials data through Materials Project. MIT - [GitHub](https://github.com/materialsproject/MPContribs) (πŸ‘¨β€πŸ’» 25 Β· πŸ”€ 19 Β· πŸ“¦ 37 Β· πŸ“‹ 98 - 20% open Β· ⏱️ 20.06.2024): ``` git clone https://github.com/materialsproject/MPContribs ``` - [PyPi](https://pypi.org/project/mpcontribs-client) (πŸ“₯ 2.2K / month): ``` pip install mpcontribs-client ```
FAIR Chemistry datasets (πŸ₯‡21 Β· ⭐ 710) - Datasets OC20, OC22, etc. Formerly known as Open Catalyst Project. MIT catalysis - [GitHub](https://github.com/FAIR-Chem/fairchem) (πŸ‘¨β€πŸ’» 39 Β· πŸ”€ 220 Β· πŸ“‹ 190 - 3% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/FAIR-Chem/fairchem ```
Open Databases Integration for Materials Design (OPTIMADE) (πŸ₯ˆ19 Β· ⭐ 73) - Specification of a common REST API for access to materials databases. CC-BY-4.0 - [GitHub](https://github.com/Materials-Consortia/OPTIMADE) (πŸ‘¨β€πŸ’» 21 Β· πŸ”€ 35 Β· πŸ“‹ 230 - 27% open Β· ⏱️ 12.06.2024): ``` git clone https://github.com/Materials-Consortia/OPTIMADE ```
QH9: A Quantum Hamiltonian Prediction Benchmark (πŸ₯ˆ12 Β· ⭐ 440) - Artificial Intelligence Research for Science (AIRS). CC-BY-NC-SA 4.0 ML-DFT - [GitHub](https://github.com/divelab/AIRS) (πŸ‘¨β€πŸ’» 28 Β· πŸ”€ 55 Β· πŸ“‹ 13 - 7% open Β· ⏱️ 14.06.2024): ``` git clone https://github.com/divelab/AIRS ```
SPICE (πŸ₯ˆ11 Β· ⭐ 130 Β· πŸ“‰) - A collection of QM data for training potential functions. MIT ML-IAP MD - [GitHub](https://github.com/openmm/spice-dataset) (πŸ”€ 6 Β· πŸ“₯ 240 Β· πŸ“‹ 57 - 24% open Β· ⏱️ 15.04.2024): ``` git clone https://github.com/openmm/spice-dataset ```
Materials Data Facility (MDF) (πŸ₯ˆ10 Β· ⭐ 10) - A simple way to publish, discover, and access materials datasets. Publication of very large datasets supported (e.g.,.. Apache-2 - [GitHub](https://github.com/materials-data-facility/connect_client) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 1 Β· ⏱️ 05.02.2024): ``` git clone https://github.com/materials-data-facility/connect_client ```
2DMD dataset (πŸ₯‰9 Β· ⭐ 5 Β· πŸ’€) - Code for Kazeev, N., Al-Maeeni, A.R., Romanov, I. et al. Sparse representation for machine learning the properties of.. Apache-2 material-defect - [GitHub](https://github.com/HSE-LAMBDA/ai4material_design) (πŸ‘¨β€πŸ’» 11 Β· πŸ”€ 3 Β· ⏱️ 21.11.2023): ``` git clone https://github.com/HSE-LAMBDA/ai4material_design ```
3DSC Database (πŸ₯‰5 Β· ⭐ 14) - Repo for the paper publishing the superconductor database with 3D crystal structures. Custom superconductors materials-discovery - [GitHub](https://github.com/aimat-lab/3DSC) (πŸ”€ 4 Β· ⏱️ 08.01.2024): ``` git clone https://github.com/aimat-lab/3DSC ```
SciGlass (πŸ₯‰5 Β· ⭐ 10 Β· πŸ’€) - The database contains a vast set of data on the properties of glass materials. MIT - [GitHub](https://github.com/drcassar/SciGlass) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 3 Β· πŸ“₯ 16 Β· ⏱️ 27.08.2023): ``` git clone https://github.com/drcassar/SciGlass ```
Show 12 hidden projects... - ATOM3D (πŸ₯ˆ18 Β· ⭐ 290 Β· πŸ’€) - ATOM3D: tasks on molecules in three dimensions. MIT biomolecules benchmarking - OpenKIM (πŸ₯ˆ10 Β· ⭐ 31 Β· πŸ’€) - The Open Knowledgebase of Interatomic Models (OpenKIM) aims to be an online resource for standardized testing, long-.. LGPL-2.1 knowledge-base pre-trained - ANI-1 Dataset (πŸ₯‰8 Β· ⭐ 93 Β· πŸ’€) - A data set of 20 million calculated off-equilibrium conformations for organic molecules. MIT - MoleculeNet Leaderboard (πŸ₯‰8 Β· ⭐ 84 Β· πŸ’€) - MIT benchmarking - GEOM (πŸ₯‰7 Β· ⭐ 180 Β· πŸ’€) - GEOM: Energy-annotated molecular conformations. Unlicensed drug-discovery - ANI-1x Datasets (πŸ₯‰6 Β· ⭐ 53 Β· πŸ’€) - The ANI-1ccx and ANI-1x data sets, coupled-cluster and density functional theory properties for organic molecules. MIT - COMP6 Benchmark dataset (πŸ₯‰6 Β· ⭐ 39 Β· πŸ’€) - COMP6 Benchmark dataset for ML potentials. MIT - OPTIMADE providers dashboard (πŸ₯‰6 Β· ⭐ 1) - A dashboard of known providers. Unlicensed - Visual Graph Datasets (πŸ₯‰5 Β· ⭐ 1) - Datasets for the training of graph neural networks (GNNs) and subsequent visualization of attributional explanations.. MIT - linear-regression-benchmarks (πŸ₯‰5 Β· ⭐ 1 Β· πŸ’€) - Data sets used for linear regression benchmarks. MIT benchmarking single-paper - paper-data-redundancy (πŸ₯‰4 Β· ⭐ 5) - Repo for the paper Exploiting redundancy in large materials datasets for efficient machine learning with less data. BSD-3 small-data single-paper - nep-data (πŸ₯‰2 Β· ⭐ 12 Β· πŸ’€) - Data related to the NEP machine-learned potential of GPUMD. Unlicensed ML-IAP MD transport-phenomena


Data Structures

Back to top

Projects that focus on providing data structures used in atomistic machine learning.

dpdata (πŸ₯‡24 Β· ⭐ 190) - Manipulating multiple atomic simulation data formats, including DeePMD-kit, VASP, LAMMPS, ABACUS, etc. LGPL-3.0 - [GitHub](https://github.com/deepmodeling/dpdata) (πŸ‘¨β€πŸ’» 60 Β· πŸ”€ 120 Β· πŸ“¦ 120 Β· πŸ“‹ 93 - 18% open Β· ⏱️ 06.06.2024): ``` git clone https://github.com/deepmodeling/dpdata ``` - [PyPi](https://pypi.org/project/dpdata) (πŸ“₯ 56K / month): ``` pip install dpdata ``` - [Conda](https://anaconda.org/deepmodeling/dpdata) (πŸ“₯ 210 Β· ⏱️ 27.09.2023): ``` conda install -c deepmodeling dpdata ```
Metatensor (πŸ₯ˆ19 Β· ⭐ 44) - Self-describing sparse tensor data format for atomistic machine learning and beyond. BSD-3 Rust C-lang C++ Python - [GitHub](https://github.com/lab-cosmo/metatensor) (πŸ‘¨β€πŸ’» 20 Β· πŸ”€ 13 Β· πŸ“₯ 22K Β· πŸ“¦ 9 Β· πŸ“‹ 180 - 34% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/lab-cosmo/metatensor ```
mp-pyrho (πŸ₯‰17 Β· ⭐ 35) - Tools for re-griding volumetric quantum chemistry data for machine-learning purposes. Custom ML-DFT - [GitHub](https://github.com/materialsproject/pyrho) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 6 Β· πŸ“¦ 22 Β· πŸ“‹ 4 - 25% open Β· ⏱️ 23.02.2024): ``` git clone https://github.com/materialsproject/pyrho ``` - [PyPi](https://pypi.org/project/mp-pyrho) (πŸ“₯ 5.2K / month): ``` pip install mp-pyrho ```
dlpack (πŸ₯‰15 Β· ⭐ 870) - common in-memory tensor structure. Apache-2 C++ - [GitHub](https://github.com/dmlc/dlpack) (πŸ‘¨β€πŸ’» 23 Β· πŸ”€ 130 Β· πŸ“‹ 67 - 37% open Β· ⏱️ 26.03.2024): ``` git clone https://github.com/dmlc/dlpack ```


Density functional theory (ML-DFT)

Back to top

Projects and models that focus on quantities of DFT, such as density functional approximations (ML-DFA), the charge density, density of states, the Hamiltonian, etc.

JAX-DFT (πŸ₯‡25 Β· ⭐ 33K) - This library provides basic building blocks that can construct DFT calculations as a differentiable program. Apache-2 - [GitHub](https://github.com/google-research/google-research) (πŸ‘¨β€πŸ’» 790 Β· πŸ”€ 7.7K Β· πŸ“‹ 1.2K - 73% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/google-research/google-research ```
MALA (πŸ₯‡20 Β· ⭐ 80) - Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data. BSD-3 - [GitHub](https://github.com/mala-project/mala) (πŸ‘¨β€πŸ’» 44 Β· πŸ”€ 23 Β· πŸ“¦ 1 Β· πŸ“‹ 260 - 11% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/mala-project/mala ```
SALTED (πŸ₯ˆ14 Β· ⭐ 26) - Symmetry-Adapted Learning of Three-dimensional Electron Densities. GPL-3.0 - [GitHub](https://github.com/andreagrisafi/SALTED) (πŸ‘¨β€πŸ’» 17 Β· πŸ”€ 4 Β· πŸ“‹ 6 - 16% open Β· ⏱️ 20.06.2024): ``` git clone https://github.com/andreagrisafi/SALTED ```
DeepH-pack (πŸ₯ˆ13 Β· ⭐ 190) - Deep neural networks for density functional theory Hamiltonian. LGPL-3.0 Julia - [GitHub](https://github.com/mzjb/DeepH-pack) (πŸ‘¨β€πŸ’» 9 Β· πŸ”€ 35 Β· πŸ“‹ 49 - 22% open Β· ⏱️ 22.05.2024): ``` git clone https://github.com/mzjb/DeepH-pack ```
QHNet (πŸ₯ˆ12 Β· ⭐ 440) - Artificial Intelligence Research for Science (AIRS). GPL-3.0 rep-learn - [GitHub](https://github.com/divelab/AIRS) (πŸ‘¨β€πŸ’» 28 Β· πŸ”€ 55 Β· πŸ“‹ 13 - 7% open Β· ⏱️ 14.06.2024): ``` git clone https://github.com/divelab/AIRS ```
Grad DFT (πŸ₯ˆ10 Β· ⭐ 70) - GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation.. Apache-2 - [GitHub](https://github.com/XanaduAI/GradDFT) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 5 Β· πŸ“‹ 54 - 20% open Β· ⏱️ 13.02.2024): ``` git clone https://github.com/XanaduAI/GradDFT ```
DeePKS-kit (πŸ₯ˆ9 Β· ⭐ 96) - a package for developing machine learning-based chemically accurate energy and density functional models. LGPL-3.0 - [GitHub](https://github.com/deepmodeling/deepks-kit) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 33 Β· πŸ“‹ 18 - 27% open Β· ⏱️ 13.04.2024): ``` git clone https://github.com/deepmodeling/deepks-kit ```
Show 18 hidden projects... - DM21 (πŸ₯‡20 Β· ⭐ 13K Β· πŸ’€) - This package provides a PySCF interface to the DM21 (DeepMind 21) family of exchange-correlation functionals described.. Apache-2 - NeuralXC (πŸ₯ˆ11 Β· ⭐ 33 Β· πŸ’€) - Implementation of a machine learned density functional. BSD-3 - ACEhamiltonians (πŸ₯ˆ11 Β· ⭐ 11 Β· πŸ’€) - Provides tools for constructing, fitting, and predicting self-consistent Hamiltonian and overlap matrices in solid-.. MIT Julia - PROPhet (πŸ₯ˆ9 Β· ⭐ 62 Β· πŸ’€) - PROPhet is a code to integrate machine learning techniques with first-principles quantum chemistry approaches. GPL-3.0 ML-IAP MD single-paper C++ - Libnxc (πŸ₯‰7 Β· ⭐ 15 Β· πŸ’€) - A library for using machine-learned exchange-correlation functionals for density-functional theory. MPL-2.0 C++ Fortran - DeepH-E3 (πŸ₯‰6 Β· ⭐ 59 Β· πŸ’€) - General framework for E(3)-equivariant neural network representation of density functional theory Hamiltonian. MIT magnetism - DeepDFT (πŸ₯‰6 Β· ⭐ 54 Β· πŸ’€) - Official implementation of DeepDFT model. MIT - Mat2Spec (πŸ₯‰6 Β· ⭐ 26 Β· πŸ’€) - MIT spectroscopy - xDeepH (πŸ₯‰5 Β· ⭐ 29 Β· πŸ’€) - Extended DeepH (xDeepH) method for magnetic materials. LGPL-3.0 magnetism Julia - ML-DFT (πŸ₯‰5 Β· ⭐ 23 Β· πŸ’€) - A package for density functional approximation using machine learning. MIT - charge-density-models (πŸ₯‰4 Β· ⭐ 10 Β· πŸ’€) - Tools to build charge density models using [fairchem](https://github.com/FAIR-Chem/fairchem). MIT rep-learn - gprep (πŸ₯‰4 Β· πŸ’€) - Fitting DFTB repulsive potentials with GPR. MIT single-paper - DeepCDP (πŸ₯‰3 Β· ⭐ 6 Β· πŸ’€) - DeepCDP: Deep learning Charge Density Prediction. Unlicensed - APET (πŸ₯‰3 Β· ⭐ 4 Β· πŸ’€) - Atomic Positional Embedding-based Transformer. GPL-3.0 density-of-states transformer - CSNN (πŸ₯‰3 Β· ⭐ 2 Β· πŸ’€) - Primary codebase of CSNN - Concentric Spherical Neural Network for 3D Representation Learning. BSD-3 - MALADA (πŸ₯‰3 Β· ⭐ 1 Β· πŸ’€) - MALA Data Acquisition: Helpful tools to build data for MALA. BSD-3 - A3MD (πŸ₯‰2 Β· ⭐ 8 Β· πŸ’€) - MPNN-like + Analytic Density Model = Accurate electron densities. Unlicensed representation-learning single-paper - kdft (πŸ₯‰1 Β· ⭐ 2 Β· πŸ’€) - The Kernel Density Functional (KDF) code allows generating ML based DFT functionals. Unlicensed


Educational Resources

Back to top

Tutorials, guides, cookbooks, recipes, etc.

πŸ”— Quantum Chemistry in the Age of Machine Learning - Book, 2022.

πŸ”— AL4MS 2023 workshop tutorials active-learning

jarvis-tools-notebooks (πŸ₯‡12 Β· ⭐ 51) - A Google-Colab Notebook Collection for Materials Design: https://jarvis.nist.gov/. NIST - [GitHub](https://github.com/JARVIS-Materials-Design/jarvis-tools-notebooks) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 24 Β· ⏱️ 02.07.2024): ``` git clone https://github.com/JARVIS-Materials-Design/jarvis-tools-notebooks ```
Deep Learning for Molecules and Materials Book (πŸ₯‡11 Β· ⭐ 590 Β· πŸ’€) - Deep learning for molecules and materials book. Custom - [GitHub](https://github.com/whitead/dmol-book) (πŸ‘¨β€πŸ’» 19 Β· πŸ”€ 110 Β· πŸ“‹ 160 - 17% open Β· ⏱️ 02.07.2023): ``` git clone https://github.com/whitead/dmol-book ```
DSECOP (πŸ₯‡11 Β· ⭐ 41) - This repository contains data science educational materials developed by DSECOP Fellows. CCO-1.0 - [GitHub](https://github.com/GDS-Education-Community-of-Practice/DSECOP) (πŸ‘¨β€πŸ’» 13 Β· πŸ”€ 25 Β· πŸ“‹ 8 - 12% open Β· ⏱️ 26.06.2024): ``` git clone https://github.com/GDS-Education-Community-of-Practice/DSECOP ```
iam-notebooks (πŸ₯ˆ10 Β· ⭐ 24) - Jupyter notebooks for the lectures of the Introduction to Atomistic Modeling. Apache-2 - [GitHub](https://github.com/ceriottm/iam-notebooks) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 5 Β· ⏱️ 26.06.2024): ``` git clone https://github.com/ceriottm/iam-notebooks ```
OPTIMADE Tutorial Exercises (πŸ₯ˆ9 Β· ⭐ 12 Β· πŸ’€) - Tutorial exercises for the OPTIMADE API. MIT datasets - [GitHub](https://github.com/Materials-Consortia/optimade-tutorial-exercises) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 7 Β· ⏱️ 27.09.2023): ``` git clone https://github.com/Materials-Consortia/optimade-tutorial-exercises ```
BestPractices (πŸ₯ˆ8 Β· ⭐ 160 Β· πŸ’€) - Things that you should (and should not) do in your Materials Informatics research. MIT - [GitHub](https://github.com/anthony-wang/BestPractices) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 68 Β· πŸ“‹ 7 - 71% open Β· ⏱️ 17.11.2023): ``` git clone https://github.com/anthony-wang/BestPractices ```
MACE-tutorials (πŸ₯‰7 Β· ⭐ 29 Β· πŸ’€) - Another set of tutorials for the MACE interatomic potential by one of the authors. MIT ML-IAP rep-learn MD - [GitHub](https://github.com/ilyes319/mace-tutorials) (πŸ”€ 8 Β· ⏱️ 10.10.2023): ``` git clone https://github.com/ilyes319/mace-tutorials ```
COSMO Software Cookbook (πŸ₯‰7 Β· ⭐ 10) - The COSMO cookbook contains recipes for atomic-scale modelling for materials and molecules. BSD-3 - [GitHub](https://github.com/lab-cosmo/software-cookbook) (πŸ‘¨β€πŸ’» 9 Β· πŸ”€ 1 Β· πŸ“‹ 12 - 16% open Β· ⏱️ 29.06.2024): ``` git clone https://github.com/lab-cosmo/software-cookbook ```
Show 14 hidden projects... - Geometric GNN Dojo (πŸ₯‡13 Β· ⭐ 440 Β· πŸ’€) - New to geometric GNNs: try our practical notebook, prepared for MPhil students at the University of Cambridge. MIT rep-learn - DeepLearningLifeSciences (πŸ₯‡11 Β· ⭐ 340 Β· πŸ’€) - Example code from the book Deep Learning for the Life Sciences. MIT - RDKit Tutorials (πŸ₯ˆ8 Β· ⭐ 240 Β· πŸ’€) - Tutorials to learn how to work with the RDKit. Custom - MAChINE (πŸ₯‰7 Β· ⭐ 1 Β· πŸ’€) - Client-Server Web App to introduce usage of ML in materials science to beginners. MIT - Applied AI for Materials (πŸ₯‰6 Β· ⭐ 55 Β· πŸ’€) - Course materials for Applied AI for Materials Science and Engineering. Unlicensed - AI4Science101 (πŸ₯‰5 Β· ⭐ 82 Β· πŸ’€) - AI for Science. Unlicensed - Machine Learning for Materials Hard and Soft (πŸ₯‰5 Β· ⭐ 34 Β· πŸ’€) - ESI-DCAFM-TACO-VDSP Summer School on Machine Learning for Materials Hard and Soft. Unlicensed - Data Handling, DoE and Statistical Analysis for Material Chemists (πŸ₯‰5 Β· ⭐ 1 Β· πŸ’€) - Notebooks for workshops of DoE course, hosted by the Computational Materials Chemistry group at Uppsala University. GPL-3.0 - ML-in-chemistry-101 (πŸ₯‰4 Β· ⭐ 65 Β· πŸ’€) - The course materials for Machine Learning in Chemistry 101. Unlicensed - chemrev-gpr (πŸ₯‰4 Β· ⭐ 6 Β· πŸ’€) - Notebooks accompanying the paper on GPR in materials and molecules in Chemical Reviews 2020. Unlicensed - MLDensity_tutorial (πŸ₯‰2 Β· ⭐ 7 Β· πŸ’€) - Tutorial files to work with ML for the charge density in molecules and solids. Unlicensed - LAMMPS-style pair potentials with GAP (πŸ₯‰2 Β· ⭐ 3 Β· πŸ’€) - A tutorial on how to create LAMMPS-style pair potentials and use them in combination with GAP potentials to run MD.. Unlicensed ML-IAP MD rep-eng - MALA Tutorial (πŸ₯‰2 Β· ⭐ 2 Β· πŸ’€) - A full MALA hands-on tutorial. Unlicensed - PiNN Lab (πŸ₯‰2 Β· ⭐ 2 Β· πŸ’€) - GPL-3.0


Explainable Artificial intelligence (XAI)

Back to top

Projects that focus on explainability and model interpretability in atomistic ML.

exmol (πŸ₯‡19 Β· ⭐ 280 Β· πŸ’€) - Explainer for black box models that predict molecule properties. MIT - [GitHub](https://github.com/ur-whitelab/exmol) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 41 Β· πŸ“¦ 20 Β· πŸ“‹ 69 - 15% open Β· ⏱️ 04.12.2023): ``` git clone https://github.com/ur-whitelab/exmol ``` - [PyPi](https://pypi.org/project/exmol) (πŸ“₯ 560 / month): ``` pip install exmol ```
MEGAN: Multi Explanation Graph Attention Student (πŸ₯ˆ6 Β· ⭐ 5) - Minimal implementation of graph attention student model architecture. MIT - [GitHub](https://github.com/aimat-lab/graph_attention_student) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 1 Β· ⏱️ 03.07.2024): ``` git clone https://github.com/aimat-lab/graph_attention_student ```
MEGAN (πŸ₯ˆ6 Β· ⭐ 5) - Minimal implementation of graph attention student model architecture. MIT XAI rep-learn - [GitHub](https://github.com/aimat-lab/graph_attention_student) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 1 Β· ⏱️ 03.07.2024): ``` git clone https://github.com/aimat-lab/graph_attention_student ```
Show 1 hidden projects... - Linear vs blackbox (πŸ₯‰3 Β· ⭐ 2 Β· πŸ’€) - Code and data related to the publication: Interpretable models for extrapolation in scientific machine learning. MIT XAI single-paper rep-eng


Electronic structure methods (ML-ESM)

Back to top

Projects and models that focus on quantities of electronic structure methods, which do not fit into either of the categories ML-WFT or ML-DFT.

Show 3 hidden projects... - QDF for molecule (πŸ₯‡9 Β· ⭐ 190 Β· πŸ’€) - Quantum deep field: data-driven wave function, electron density generation, and energy prediction and extrapolation.. MIT - halex (πŸ₯ˆ4 Β· ⭐ 2) - Hamiltonian Learning for Excited States https://doi.org/10.48550/arXiv.2311.00844. Unlicensed excited-states - e3psi (πŸ₯‰3 Β· ⭐ 3) - Equivariant machine learning library for learning from electronic structures. LGPL-3.0


General Tools

Back to top

General tools for atomistic machine learning.

DeepChem (πŸ₯‡36 Β· ⭐ 5.3K) - Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, Materials Science and Biology. MIT - [GitHub](https://github.com/deepchem/deepchem) (πŸ‘¨β€πŸ’» 240 Β· πŸ”€ 1.6K Β· πŸ“¦ 390 Β· πŸ“‹ 1.7K - 26% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/deepchem/deepchem ``` - [PyPi](https://pypi.org/project/deepchem) (πŸ“₯ 22K / month): ``` pip install deepchem ``` - [Conda](https://anaconda.org/conda-forge/deepchem) (πŸ“₯ 110K Β· ⏱️ 05.04.2024): ``` conda install -c conda-forge deepchem ``` - [Docker Hub](https://hub.docker.com/r/deepchemio/deepchem) (πŸ“₯ 7.2K Β· ⭐ 5 Β· ⏱️ 04.07.2024): ``` docker pull deepchemio/deepchem ```
RDKit (πŸ₯‡32 Β· ⭐ 2.5K) - BSD-3 C++ - [GitHub](https://github.com/rdkit/rdkit) (πŸ‘¨β€πŸ’» 230 Β· πŸ”€ 820 Β· πŸ“₯ 1.3K Β· πŸ“¦ 3 Β· πŸ“‹ 3.2K - 28% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/rdkit/rdkit ``` - [PyPi](https://pypi.org/project/rdkit) (πŸ“₯ 440K / month): ``` pip install rdkit ``` - [Conda](https://anaconda.org/rdkit/rdkit) (πŸ“₯ 2.6M Β· ⏱️ 16.06.2023): ``` conda install -c rdkit rdkit ```
Matminer (πŸ₯‡28 Β· ⭐ 440) - Data mining for materials science. Custom - [GitHub](https://github.com/hackingmaterials/matminer) (πŸ‘¨β€πŸ’» 54 Β· πŸ”€ 180 Β· πŸ“¦ 300 Β· πŸ“‹ 220 - 10% open Β· ⏱️ 27.05.2024): ``` git clone https://github.com/hackingmaterials/matminer ``` - [PyPi](https://pypi.org/project/matminer) (πŸ“₯ 8.5K / month): ``` pip install matminer ``` - [Conda](https://anaconda.org/conda-forge/matminer) (πŸ“₯ 64K Β· ⏱️ 28.03.2024): ``` conda install -c conda-forge matminer ```
QUIP (πŸ₯ˆ25 Β· ⭐ 340 Β· πŸ“ˆ) - libAtoms/QUIP molecular dynamics framework: https://libatoms.github.io. GPL-2.0 MD ML-IAP rep-eng Fortran - [GitHub](https://github.com/libAtoms/QUIP) (πŸ‘¨β€πŸ’» 83 Β· πŸ”€ 120 Β· πŸ“₯ 490 Β· πŸ“¦ 38 Β· πŸ“‹ 460 - 22% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/libAtoms/QUIP ``` - [PyPi](https://pypi.org/project/quippy-ase) (πŸ“₯ 3.2K / month): ``` pip install quippy-ase ``` - [Docker Hub](https://hub.docker.com/r/libatomsquip/quip) (πŸ“₯ 9.9K Β· ⭐ 4 Β· ⏱️ 24.04.2023): ``` docker pull libatomsquip/quip ```
MAML (πŸ₯ˆ24 Β· ⭐ 340) - Python for Materials Machine Learning, Materials Descriptors, Machine Learning Force Fields, Deep Learning, etc. BSD-3 - [GitHub](https://github.com/materialsvirtuallab/maml) (πŸ‘¨β€πŸ’» 32 Β· πŸ”€ 73 Β· πŸ“¦ 9 Β· πŸ“‹ 68 - 10% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/materialsvirtuallab/maml ``` - [PyPi](https://pypi.org/project/maml) (πŸ“₯ 310 / month): ``` pip install maml ```
JARVIS-Tools (πŸ₯ˆ24 Β· ⭐ 280) - JARVIS-Tools: an open-source software package for data-driven atomistic materials design. Publications:.. Custom - [GitHub](https://github.com/usnistgov/jarvis) (πŸ‘¨β€πŸ’» 15 Β· πŸ”€ 120 Β· πŸ“¦ 90 Β· πŸ“‹ 89 - 49% open Β· ⏱️ 28.06.2024): ``` git clone https://github.com/usnistgov/jarvis ``` - [PyPi](https://pypi.org/project/jarvis-tools) (πŸ“₯ 18K / month): ``` pip install jarvis-tools ``` - [Conda](https://anaconda.org/conda-forge/jarvis-tools) (πŸ“₯ 69K Β· ⏱️ 28.06.2024): ``` conda install -c conda-forge jarvis-tools ```
MAST-ML (πŸ₯ˆ20 Β· ⭐ 97) - MAterials Simulation Toolkit for Machine Learning (MAST-ML). MIT - [GitHub](https://github.com/uw-cmg/MAST-ML) (πŸ‘¨β€πŸ’» 18 Β· πŸ”€ 56 Β· πŸ“₯ 93 Β· πŸ“¦ 43 Β· πŸ“‹ 210 - 10% open Β· ⏱️ 17.04.2024): ``` git clone https://github.com/uw-cmg/MAST-ML ```
Scikit-Matter (πŸ₯ˆ19 Β· ⭐ 70) - A collection of scikit-learn compatible utilities that implement methods born out of the materials science and.. BSD-3 scikit-learn - [GitHub](https://github.com/scikit-learn-contrib/scikit-matter) (πŸ‘¨β€πŸ’» 15 Β· πŸ”€ 18 Β· πŸ“¦ 10 Β· πŸ“‹ 69 - 18% open Β· ⏱️ 15.06.2024): ``` git clone https://github.com/scikit-learn-contrib/scikit-matter ``` - [PyPi](https://pypi.org/project/skmatter) (πŸ“₯ 2K / month): ``` pip install skmatter ``` - [Conda](https://anaconda.org/conda-forge/skmatter) (πŸ“₯ 1.8K Β· ⏱️ 24.08.2023): ``` conda install -c conda-forge skmatter ```
XenonPy (πŸ₯‰14 Β· ⭐ 130) - XenonPy is a Python Software for Materials Informatics. BSD-3 - [GitHub](https://github.com/yoshida-lab/XenonPy) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 57 Β· πŸ“₯ 1.4K Β· πŸ“‹ 85 - 22% open Β· ⏱️ 21.04.2024): ``` git clone https://github.com/yoshida-lab/XenonPy ``` - [PyPi](https://pypi.org/project/xenonpy) (πŸ“₯ 220 / month): ``` pip install xenonpy ```
Artificial Intelligence for Science (AIRS) (πŸ₯‰12 Β· ⭐ 440) - Artificial Intelligence Research for Science (AIRS). GPL-3.0 license rep-learn generative ML-IAP MD ML-DFT ML-WFT biomolecules - [GitHub](https://github.com/divelab/AIRS) (πŸ‘¨β€πŸ’» 28 Β· πŸ”€ 55 Β· πŸ“‹ 13 - 7% open Β· ⏱️ 14.06.2024): ``` git clone https://github.com/divelab/AIRS ```
AMPtorch (πŸ₯‰11 Β· ⭐ 58 Β· πŸ’€) - AMPtorch: Atomistic Machine Learning Package (AMP) - PyTorch. GPL-3.0 - [GitHub](https://github.com/ulissigroup/amptorch) (πŸ‘¨β€πŸ’» 14 Β· πŸ”€ 32 Β· πŸ“‹ 33 - 21% open Β· ⏱️ 16.07.2023): ``` git clone https://github.com/ulissigroup/amptorch ```
Equisolve (πŸ₯‰6 Β· ⭐ 5 Β· πŸ’€) - A ML toolkit package utilizing the metatensor data format to build models for the prediction of equivariant properties.. BSD-3 ML-IAP - [GitHub](https://github.com/lab-cosmo/equisolve) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 1 Β· πŸ“‹ 23 - 82% open Β· ⏱️ 27.10.2023): ``` git clone https://github.com/lab-cosmo/equisolve ```
Show 10 hidden projects... - QML (πŸ₯ˆ15 Β· ⭐ 200 Β· πŸ’€) - QML: Quantum Machine Learning. MIT - Automatminer (πŸ₯‰14 Β· ⭐ 130 Β· πŸ’€) - An automatic engine for predicting materials properties. Custom - OpenChem (πŸ₯‰10 Β· ⭐ 670 Β· πŸ’€) - OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research. MIT - JAXChem (πŸ₯‰7 Β· ⭐ 74 Β· πŸ’€) - JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling. MIT - uncertainty_benchmarking (πŸ₯‰7 Β· ⭐ 37 Β· πŸ’€) - Various code/notebooks to benchmark different ways we could estimate uncertainty in ML predictions. Unlicensed benchmarking probabilistic - torchchem (πŸ₯‰7 Β· ⭐ 34 Β· πŸ’€) - An experimental repo for experimenting with PyTorch models. MIT - ACEatoms (πŸ₯‰4 Β· ⭐ 2 Β· πŸ’€) - Generic code for modelling atomic properties using ACE. Custom Julia - MLatom (πŸ₯‰4) - Machine learning for atomistic simulations. Custom - Magpie (πŸ₯‰3) - Materials Agnostic Platform for Informatics and Exploration (Magpie). MIT Java - quantum-structure-ml (πŸ₯‰2 Β· ⭐ 2 Β· πŸ’€) - Multi-class classification model for predicting the magnetic order of magnetic structures and a binary classification.. Unlicensed magnetism benchmarking


Generative Models

Back to top

Projects that implement generative models for atomistic ML.

GT4SD (πŸ₯‡19 Β· ⭐ 320) - GT4SD, an open-source library to accelerate hypothesis generation in the scientific discovery process. MIT pre-trained drug-discovery rep-learn - [GitHub](https://github.com/GT4SD/gt4sd-core) (πŸ‘¨β€πŸ’» 20 Β· πŸ”€ 66 Β· πŸ“‹ 99 - 1% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/GT4SD/gt4sd-core ``` - [PyPi](https://pypi.org/project/gt4sd) (πŸ“₯ 510 / month): ``` pip install gt4sd ```
MoLeR (πŸ₯‡15 Β· ⭐ 250) - Implementation of MoLeR: a generative model of molecular graphs which supports scaffold-constrained generation. MIT - [GitHub](https://github.com/microsoft/molecule-generation) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 36 Β· πŸ“‹ 37 - 24% open Β· ⏱️ 03.01.2024): ``` git clone https://github.com/microsoft/molecule-generation ``` - [PyPi](https://pypi.org/project/molecule-generation) (πŸ“₯ 220 / month): ``` pip install molecule-generation ```
SchNetPack G-SchNet (πŸ₯ˆ14 Β· ⭐ 42 Β· πŸ“ˆ) - G-SchNet extension for SchNetPack. MIT - [GitHub](https://github.com/atomistic-machine-learning/schnetpack-gschnet) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 8 Β· ⏱️ 03.07.2024): ``` git clone https://github.com/atomistic-machine-learning/schnetpack-gschnet ```
bVAE-IM (πŸ₯‰8 Β· ⭐ 11 Β· πŸ’€) - Implementation of Chemical Design with GPU-based Ising Machine. MIT QML single-paper - [GitHub](https://github.com/tsudalab/bVAE-IM) (πŸ”€ 3 Β· ⏱️ 11.07.2023): ``` git clone https://github.com/tsudalab/bVAE-IM ```
COATI (πŸ₯‰5 Β· ⭐ 83) - COATI: multi-modal contrastive pre-training for representing and traversing chemical space. Apache-2 drug-discovery pre-trained rep-learn - [GitHub](https://github.com/terraytherapeutics/COATI) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 5 Β· πŸ“‹ 3 - 33% open Β· ⏱️ 23.03.2024): ``` git clone https://github.com/terraytherapeutics/COATI ```
Show 6 hidden projects... - synspace (πŸ₯ˆ12 Β· ⭐ 35 Β· πŸ’€) - Synthesis generative model. MIT - EDM (πŸ₯ˆ10 Β· ⭐ 410 Β· πŸ’€) - E(3) Equivariant Diffusion Model for Molecule Generation in 3D. MIT - G-SchNet (πŸ₯‰8 Β· ⭐ 130 Β· πŸ’€) - G-SchNet - a generative model for 3d molecular structures. MIT - cG-SchNet (πŸ₯‰8 Β· ⭐ 46 Β· πŸ’€) - cG-SchNet - a conditional generative neural network for 3d molecular structures. MIT - rxngenerator (πŸ₯‰5 Β· ⭐ 11 Β· πŸ’€) - A generative model for molecular generation via multi-step chemical reactions. MIT - MolSLEPA (πŸ₯‰5 Β· ⭐ 5 Β· πŸ’€) - Interpretable Fragment-based Molecule Design with Self-learning Entropic Population Annealing. MIT XAI


Interatomic Potentials (ML-IAP)

Back to top

Machine learning interatomic potentials (aka ML-IAP, MLIAP, MLIP, MLP) and force fields (ML-FF) for molecular dynamics.

DeePMD-kit (πŸ₯‡28 Β· ⭐ 1.4K Β· πŸ“ˆ) - A deep learning package for many-body potential energy representation and molecular dynamics. LGPL-3.0 C++ - [GitHub](https://github.com/deepmodeling/deepmd-kit) (πŸ‘¨β€πŸ’» 69 Β· πŸ”€ 470 Β· πŸ“₯ 36K Β· πŸ“¦ 16 Β· πŸ“‹ 720 - 9% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/deepmodeling/deepmd-kit ``` - [PyPi](https://pypi.org/project/deepmd-kit) (πŸ“₯ 1.1K / month): ``` pip install deepmd-kit ``` - [Conda](https://anaconda.org/deepmodeling/deepmd-kit) (πŸ“₯ 1.1K Β· ⏱️ 06.04.2024): ``` conda install -c deepmodeling deepmd-kit ``` - [Docker Hub](https://hub.docker.com/r/deepmodeling/deepmd-kit) (πŸ“₯ 2.3K Β· ⭐ 1 Β· ⏱️ 02.07.2024): ``` docker pull deepmodeling/deepmd-kit ```
NequIP (πŸ₯‡24 Β· ⭐ 560) - NequIP is a code for building E(3)-equivariant interatomic potentials. MIT - [GitHub](https://github.com/mir-group/nequip) (πŸ‘¨β€πŸ’» 11 Β· πŸ”€ 120 Β· πŸ“¦ 18 Β· πŸ“‹ 82 - 28% open Β· ⏱️ 10.05.2024): ``` git clone https://github.com/mir-group/nequip ``` - [PyPi](https://pypi.org/project/nequip) (πŸ“₯ 2.6K / month): ``` pip install nequip ``` - [Conda](https://anaconda.org/conda-forge/nequip) (πŸ“₯ 4.8K Β· ⏱️ 11.05.2024): ``` conda install -c conda-forge nequip ```
TorchANI (πŸ₯‡22 Β· ⭐ 440 Β· πŸ’€) - Accurate Neural Network Potential on PyTorch. MIT - [GitHub](https://github.com/aiqm/torchani) (πŸ‘¨β€πŸ’» 17 Β· πŸ”€ 120 Β· πŸ“¦ 39 Β· πŸ“‹ 170 - 13% open Β· ⏱️ 14.11.2023): ``` git clone https://github.com/aiqm/torchani ``` - [PyPi](https://pypi.org/project/torchani) (πŸ“₯ 4.6K / month): ``` pip install torchani ``` - [Conda](https://anaconda.org/conda-forge/torchani) (πŸ“₯ 350K Β· ⏱️ 31.05.2024): ``` conda install -c conda-forge torchani ```
MACE (πŸ₯‡22 Β· ⭐ 410) - MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing. MIT - [GitHub](https://github.com/ACEsuit/mace) (πŸ‘¨β€πŸ’» 32 Β· πŸ”€ 150 Β· πŸ“‹ 200 - 20% open Β· ⏱️ 26.06.2024): ``` git clone https://github.com/ACEsuit/mace ```
CHGNet (πŸ₯‡22 Β· ⭐ 210) - Pretrained universal neural network potential for charge-informed atomistic modeling https://chgnet.lbl.gov. Custom MD pre-trained electrostatics magnetism structure-relaxation - [GitHub](https://github.com/CederGroupHub/chgnet) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 54 Β· πŸ“¦ 24 Β· πŸ“‹ 49 - 6% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/CederGroupHub/chgnet ``` - [PyPi](https://pypi.org/project/chgnet) (πŸ“₯ 24K / month): ``` pip install chgnet ```
GPUMD (πŸ₯‡21 Β· ⭐ 410 Β· πŸ“‰) - GPUMD is a highly efficient general-purpose molecular dynamic (MD) package and enables machine-learned potentials.. GPL-3.0 MD C++ electrostatics - [GitHub](https://github.com/brucefan1983/GPUMD) (πŸ‘¨β€πŸ’» 29 Β· πŸ”€ 110 Β· πŸ“‹ 170 - 11% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/brucefan1983/GPUMD ```
TorchMD-NET (πŸ₯‡21 Β· ⭐ 290) - Training neural network potentials. MIT MD rep-learn transformer pre-trained - [GitHub](https://github.com/torchmd/torchmd-net) (πŸ‘¨β€πŸ’» 16 Β· πŸ”€ 67 Β· πŸ“‹ 100 - 19% open Β· ⏱️ 28.06.2024): ``` git clone https://github.com/torchmd/torchmd-net ``` - [Conda](https://anaconda.org/conda-forge/torchmd-net) (πŸ“₯ 69K Β· ⏱️ 03.06.2024): ``` conda install -c conda-forge torchmd-net ```
DP-GEN (πŸ₯‡21 Β· ⭐ 280 Β· πŸ“‰) - The deep potential generator to generate a deep-learning based model of interatomic potential energy and force field. LGPL-3.0 workflows - [GitHub](https://github.com/deepmodeling/dpgen) (πŸ‘¨β€πŸ’» 64 Β· πŸ”€ 170 Β· πŸ“₯ 1.7K Β· πŸ“¦ 6 Β· πŸ“‹ 280 - 9% open Β· ⏱️ 10.04.2024): ``` git clone https://github.com/deepmodeling/dpgen ``` - [PyPi](https://pypi.org/project/dpgen) (πŸ“₯ 370 / month): ``` pip install dpgen ``` - [Conda](https://anaconda.org/deepmodeling/dpgen) (πŸ“₯ 200 Β· ⏱️ 16.06.2023): ``` conda install -c deepmodeling dpgen ```
fairchem (πŸ₯ˆ19 Β· ⭐ 710) - FAIR Chemistrys library of machine learning methods for chemistry. Formerly known as Open Catalyst Project (ocp). Unlicensed pre-trained rep-learn catalysis - [GitHub](https://github.com/FAIR-Chem/fairchem) (πŸ‘¨β€πŸ’» 39 Β· πŸ”€ 220 Β· πŸ“‹ 190 - 3% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/FAIR-Chem/fairchem ```
apax (πŸ₯ˆ16 Β· ⭐ 12) - A flexible and performant framework for training machine learning potentials. MIT - [GitHub](https://github.com/apax-hub/apax) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 1 Β· πŸ“¦ 2 Β· πŸ“‹ 110 - 17% open Β· ⏱️ 17.05.2024): ``` git clone https://github.com/apax-hub/apax ``` - [PyPi](https://pypi.org/project/apax) (πŸ“₯ 47 / month): ``` pip install apax ```
sGDML (πŸ₯ˆ15 Β· ⭐ 140 Β· πŸ’€) - sGDML - Reference implementation of the Symmetric Gradient Domain Machine Learning model. MIT - [GitHub](https://github.com/stefanch/sGDML) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 36 Β· πŸ“¦ 9 Β· πŸ“‹ 17 - 35% open Β· ⏱️ 31.08.2023): ``` git clone https://github.com/stefanch/sGDML ``` - [PyPi](https://pypi.org/project/sgdml) (πŸ“₯ 160 / month): ``` pip install sgdml ```
Ultra-Fast Force Fields (UF3) (πŸ₯ˆ15 Β· ⭐ 55 Β· πŸ“ˆ) - UF3: a python library for generating ultra-fast interatomic potentials. Apache-2 - [GitHub](https://github.com/uf3/uf3) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 20 Β· πŸ“¦ 1 Β· πŸ“‹ 41 - 29% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/uf3/uf3 ``` - [PyPi](https://pypi.org/project/uf3) (πŸ“₯ 30 / month): ``` pip install uf3 ```
KLIFF (πŸ₯ˆ15 Β· ⭐ 34) - KIM-based Learning-Integrated Fitting Framework for interatomic potentials. LGPL-2.1 probabilistic workflows - [GitHub](https://github.com/openkim/kliff) (πŸ‘¨β€πŸ’» 9 Β· πŸ”€ 20 Β· πŸ“¦ 3 Β· πŸ“‹ 40 - 52% open Β· ⏱️ 28.06.2024): ``` git clone https://github.com/openkim/kliff ``` - [PyPi](https://pypi.org/project/kliff) (πŸ“₯ 49 / month): ``` pip install kliff ``` - [Conda](https://anaconda.org/conda-forge/kliff) (πŸ“₯ 86K Β· ⏱️ 18.12.2023): ``` conda install -c conda-forge kliff ```
wfl (πŸ₯ˆ15 Β· ⭐ 22) - Workflow is a Python toolkit for building interatomic potential creation and atomistic simulation workflows. GPL-2.0 workflows HTC - [GitHub](https://github.com/libAtoms/workflow) (πŸ‘¨β€πŸ’» 17 Β· πŸ”€ 15 Β· πŸ“¦ 1 Β· πŸ“‹ 150 - 42% open Β· ⏱️ 14.06.2024): ``` git clone https://github.com/libAtoms/workflow ```
PyXtalFF (πŸ₯ˆ14 Β· ⭐ 84) - Machine Learning Interatomic Potential Predictions. MIT - [GitHub](https://github.com/MaterSim/PyXtal_FF) (πŸ‘¨β€πŸ’» 9 Β· πŸ”€ 23 Β· πŸ“‹ 62 - 17% open Β· ⏱️ 07.01.2024): ``` git clone https://github.com/MaterSim/PyXtal_FF ``` - [PyPi](https://pypi.org/project/pyxtal_ff) (πŸ“₯ 60 / month): ``` pip install pyxtal_ff ```
NNPOps (πŸ₯ˆ14 Β· ⭐ 79 Β· πŸ’€) - High-performance operations for neural network potentials. MIT MD C++ - [GitHub](https://github.com/openmm/NNPOps) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 16 Β· πŸ“‹ 55 - 38% open Β· ⏱️ 25.07.2023): ``` git clone https://github.com/openmm/NNPOps ``` - [Conda](https://anaconda.org/conda-forge/nnpops) (πŸ“₯ 150K Β· ⏱️ 31.05.2024): ``` conda install -c conda-forge nnpops ```
CCS_fit (πŸ₯ˆ13 Β· ⭐ 7) - Curvature Constrained Splines. GPL-3.0 - [GitHub](https://github.com/Teoroo-CMC/CCS) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 10 Β· πŸ“₯ 400 Β· πŸ“‹ 14 - 57% open Β· ⏱️ 16.02.2024): ``` git clone https://github.com/Teoroo-CMC/CCS ``` - [PyPi](https://pypi.org/project/ccs_fit) (πŸ“₯ 140 / month): ``` pip install ccs_fit ```
ANI-1 (πŸ₯ˆ12 Β· ⭐ 220) - ANI-1 neural net potential with python interface (ASE). MIT - [GitHub](https://github.com/isayev/ASE_ANI) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 55 Β· πŸ“‹ 37 - 43% open Β· ⏱️ 11.03.2024): ``` git clone https://github.com/isayev/ASE_ANI ```
Neural Force Field (πŸ₯ˆ12 Β· ⭐ 220) - Neural Network Force Field based on PyTorch. MIT pre-trained - [GitHub](https://github.com/learningmatter-mit/NeuralForceField) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 46 Β· ⏱️ 04.07.2024): ``` git clone https://github.com/learningmatter-mit/NeuralForceField ```
DMFF (πŸ₯ˆ12 Β· ⭐ 140) - DMFF (Differentiable Molecular Force Field) is a Jax-based python package that provides a full differentiable.. LGPL-3.0 - [GitHub](https://github.com/deepmodeling/DMFF) (πŸ‘¨β€πŸ’» 14 Β· πŸ”€ 40 Β· πŸ“‹ 26 - 38% open Β· ⏱️ 12.01.2024): ``` git clone https://github.com/deepmodeling/DMFF ```
PiNN (πŸ₯ˆ11 Β· ⭐ 100) - A Python library for building atomic neural networks. BSD-3 - [GitHub](https://github.com/Teoroo-CMC/PiNN) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 30 Β· πŸ“‹ 6 - 16% open Β· ⏱️ 27.06.2024): ``` git clone https://github.com/Teoroo-CMC/PiNN ``` - [Docker Hub](https://hub.docker.com/r/teoroo/pinn) (πŸ“₯ 240 Β· ⏱️ 27.06.2024): ``` docker pull teoroo/pinn ```
Pacemaker (πŸ₯ˆ11 Β· ⭐ 57) - Python package for fitting atomic cluster expansion (ACE) potentials. Custom - [GitHub](https://github.com/ICAMS/python-ace) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 15 Β· πŸ“‹ 44 - 27% open Β· ⏱️ 16.02.2024): ``` git clone https://github.com/ICAMS/python-ace ``` - [PyPi](https://pypi.org/project/python-ace) (πŸ“₯ 17 / month): ``` pip install python-ace ```
So3krates (MLFF) (πŸ₯ˆ11 Β· ⭐ 55) - Build neural networks for machine learning force fields with JAX. MIT - [GitHub](https://github.com/thorben-frank/mlff) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 12 Β· πŸ“‹ 9 - 55% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/thorben-frank/mlff ```
tinker-hp (πŸ₯‰10 Β· ⭐ 77) - Tinker-HP: High-Performance Massively Parallel Evolution of Tinker on CPUs & GPUs. Custom - [GitHub](https://github.com/TinkerTools/tinker-hp) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 21 Β· πŸ“‹ 19 - 15% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/TinkerTools/tinker-hp ```
Point Edge Transformer (PET) (πŸ₯‰10 Β· ⭐ 14) - Point Edge Transformer. MIT rep-learn transformer - [GitHub](https://github.com/spozdn/pet) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 5 Β· ⏱️ 02.07.2024): ``` git clone https://github.com/spozdn/pet ```
DimeNet (πŸ₯‰9 Β· ⭐ 280 Β· πŸ’€) - DimeNet and DimeNet++ models, as proposed in Directional Message Passing for Molecular Graphs (ICLR 2020) and Fast and.. Custom - [GitHub](https://github.com/gasteigerjo/dimenet) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 57 Β· πŸ“¦ 1 Β· πŸ“‹ 31 - 3% open Β· ⏱️ 03.10.2023): ``` git clone https://github.com/gasteigerjo/dimenet ```
TurboGAP (πŸ₯‰9 Β· ⭐ 16) - The TurboGAP code. Custom Fortran - [GitHub](https://github.com/mcaroba/turbogap) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 8 Β· πŸ“‹ 8 - 62% open Β· ⏱️ 18.06.2024): ``` git clone https://github.com/mcaroba/turbogap ```
ACEfit (πŸ₯‰9 Β· ⭐ 8) - MIT Julia - [GitHub](https://github.com/ACEsuit/ACEfit.jl) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 5 Β· πŸ“‹ 55 - 40% open Β· ⏱️ 28.03.2024): ``` git clone https://github.com/ACEsuit/ACEfit.jl ```
MACE-Jax (πŸ₯‰8 Β· ⭐ 51 Β· πŸ’€) - Equivariant machine learning interatomic potentials in JAX. MIT - [GitHub](https://github.com/ACEsuit/mace-jax) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 3 Β· πŸ“‹ 5 - 40% open Β· ⏱️ 04.10.2023): ``` git clone https://github.com/ACEsuit/mace-jax ```
PyNEP (πŸ₯‰8 Β· ⭐ 40) - A python interface of the machine learning potential NEP used in GPUMD. MIT - [GitHub](https://github.com/bigd4/PyNEP) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 16 Β· πŸ“‹ 11 - 36% open Β· ⏱️ 01.06.2024): ``` git clone https://github.com/bigd4/PyNEP ```
GAP (πŸ₯‰8 Β· ⭐ 39) - Gaussian Approximation Potential (GAP). Custom - [GitHub](https://github.com/libAtoms/GAP) (πŸ‘¨β€πŸ’» 13 Β· πŸ”€ 20 Β· ⏱️ 03.07.2024): ``` git clone https://github.com/libAtoms/GAP ```
ACE1.jl (πŸ₯‰8 Β· ⭐ 20) - Atomic Cluster Expansion for Modelling Invariant Atomic Properties. Custom Julia - [GitHub](https://github.com/ACEsuit/ACE1.jl) (πŸ‘¨β€πŸ’» 9 Β· πŸ”€ 7 Β· πŸ“‹ 46 - 47% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/ACEsuit/ACE1.jl ```
ALF (πŸ₯‰7 Β· ⭐ 26) - A framework for performing active learning for training machine-learned interatomic potentials. Custom active-learning - [GitHub](https://github.com/lanl/ALF) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 11 Β· ⏱️ 28.05.2024): ``` git clone https://github.com/lanl/alf ```
MLXDM (πŸ₯‰7 Β· ⭐ 5) - A Neural Network Potential with Rigorous Treatment of Long-Range Dispersion https://doi.org/10.1039/D2DD00150K. MIT long-range - [GitHub](https://github.com/RowleyGroup/MLXDM) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 2 Β· ⏱️ 03.07.2024): ``` git clone https://github.com/RowleyGroup/MLXDM ```
ACE1Pack.jl (πŸ₯‰6 Β· πŸ’€) - Provides convenience functionality for the usage of ACE1.jl, ACEfit.jl, JuLIP.jl for fitting interatomic potentials.. MIT Julia - [GitHub](https://github.com/ACEsuit/ACE1pack.jl) (πŸ‘¨β€πŸ’» 11 Β· ⏱️ 21.08.2023): ``` git clone https://github.com/ACEsuit/ACE1pack.jl ```
Allegro-JAX ( ⭐ 16) - JAX implementation of the Allegro interatomic potential. Unlicensed - [GitHub](https://github.com/mariogeiger/allegro-jax) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 2 Β· ⏱️ 09.04.2024): ``` git clone https://github.com/mariogeiger/allegro-jax ```
Show 29 hidden projects... - MEGNet (πŸ₯‡22 Β· ⭐ 490 Β· πŸ’€) - Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals. BSD-3 - M3GNet (πŸ₯ˆ17 Β· ⭐ 220 Β· πŸ’€) - Materials graph network with 3-body interactions featuring a DFT surrogate crystal relaxer and a state-of-the-art.. BSD-3 - n2p2 (πŸ₯ˆ13 Β· ⭐ 210 Β· πŸ’€) - n2p2 - A Neural Network Potential Package. GPL-3.0 C++ - TensorMol (πŸ₯ˆ12 Β· ⭐ 270 Β· πŸ’€) - Tensorflow + Molecules = TensorMol. GPL-3.0 single-paper - SIMPLE-NN (πŸ₯ˆ11 Β· ⭐ 47 Β· πŸ’€) - SIMPLE-NN(SNU Interatomic Machine-learning PotentiaL packagE version Neural Network). GPL-3.0 - Allegro (πŸ₯‰10 Β· ⭐ 300 Β· πŸ’€) - Allegro is an open-source code for building highly scalable and accurate equivariant deep learning interatomic.. MIT - NNsforMD (πŸ₯‰10 Β· ⭐ 10 Β· πŸ’€) - Neural network class for molecular dynamics to predict potential energy, forces and non-adiabatic couplings. MIT - SchNet (πŸ₯‰9 Β· ⭐ 210 Β· πŸ’€) - SchNet - a deep learning architecture for quantum chemistry. MIT - GemNet (πŸ₯‰9 Β· ⭐ 170 Β· πŸ’€) - GemNet model in PyTorch, as proposed in GemNet: Universal Directional Graph Neural Networks for Molecules (NeurIPS.. Custom - ACE.jl (πŸ₯‰9 Β· ⭐ 65 Β· πŸ’€) - Parameterisation of Equivariant Properties of Particle Systems. Custom Julia - AIMNet (πŸ₯‰8 Β· ⭐ 92 Β· πŸ’€) - Atoms In Molecules Neural Network Potential. MIT single-paper - SNAP (πŸ₯‰8 Β· ⭐ 36 Β· πŸ’€) - Repository for spectral neighbor analysis potential (SNAP) model development. BSD-3 - Atomistic Adversarial Attacks (πŸ₯‰8 Β· ⭐ 29 Β· πŸ’€) - Code for performing adversarial attacks on atomistic systems using NN potentials. MIT probabilistic - calorine (πŸ₯‰8 Β· ⭐ 12 Β· πŸ’€) - A Python package for constructing and sampling neuroevolution potential models. https://doi.org/10.21105/joss.06264. Custom - PhysNet (πŸ₯‰7 Β· ⭐ 89 Β· πŸ’€) - Code for training PhysNet models. MIT electrostatics - SIMPLE-NN v2 (πŸ₯‰7 Β· ⭐ 38 Β· πŸ’€) - GPL-3.0 - MLIP-3 (πŸ₯‰6 Β· ⭐ 25 Β· πŸ’€) - MLIP-3: Active learning on atomic environments with Moment Tensor Potentials (MTP). BSD-2 C++ - testing-framework (πŸ₯‰6 Β· ⭐ 11 Β· πŸ’€) - The purpose of this repository is to aid the testing of a large number of interatomic potentials for a variety of.. Unlicensed benchmarking - PANNA (πŸ₯‰6 Β· ⭐ 9 Β· πŸ’€) - A package to train and validate all-to-all connected network models for BP[1] and modified-BP[2] type local atomic.. MIT benchmarking - Alchemical learning (πŸ₯‰5 Β· ⭐ 2 Β· πŸ’€) - Code for the Modeling high-entropy transition metal alloys with alchemical compression article. BSD-3 - glp (πŸ₯‰4 Β· ⭐ 17) - tools for graph-based machine-learning potentials in jax. MIT - NequIP-JAX (πŸ₯‰4 Β· ⭐ 16 Β· πŸ’€) - JAX implementation of the NequIP interatomic potential. Unlicensed - TensorPotential (πŸ₯‰4 Β· ⭐ 7 Β· πŸ’€) - Tensorpotential is a TensorFlow based tool for development, fitting ML interatomic potentials from electronic.. Custom - ACE Workflows (πŸ₯‰4 Β· πŸ’€) - Workflow Examples for ACE Models. Unlicensed Julia workflows - PeriodicPotentials (πŸ₯‰4 Β· πŸ’€) - A Periodic table app that displays potentials based on the selected elements. MIT community-resource viz JavaScript - MEGNetSparse (πŸ₯‰3 Β· ⭐ 1 Β· πŸ’€) - A library imlementing a graph neural network with sparse representation from Code for Kazeev, N., Al-Maeeni, A.R.,.. MIT material-defect - SingleNN (πŸ₯‰2 Β· ⭐ 8 Β· πŸ’€) - An efficient package for training and executing neural-network interatomic potentials. Unlicensed C++ - RuNNer (πŸ₯‰2) - The RuNNer Neural Network Energy Representation is a Fortran-based framework for the construction of Behler-.. GPL-3.0 Fortran - mlp (πŸ₯‰1 Β· ⭐ 1 Β· πŸ’€) - Proper orthogonal descriptors for efficient and accurate interatomic potentials... Unlicensed Julia


Language Models

Back to top

Projects that use (large) language models (LMs, LLMs) or natural language procesing (NLP) techniques for atomistic ML.

paper-qa (πŸ₯‡26 Β· ⭐ 3.8K) - LLM Chain for answering questions from documents with citations. Apache-2 - [GitHub](https://github.com/whitead/paper-qa) (πŸ‘¨β€πŸ’» 17 Β· πŸ”€ 350 Β· πŸ“¦ 62 Β· πŸ“‹ 140 - 48% open Β· ⏱️ 28.06.2024): ``` git clone https://github.com/whitead/paper-qa ``` - [PyPi](https://pypi.org/project/paper-qa) (πŸ“₯ 4.6K / month): ``` pip install paper-qa ```
ChemCrow (πŸ₯‡16 Β· ⭐ 510) - Chemcrow. MIT - [GitHub](https://github.com/ur-whitelab/chemcrow-public) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 69 Β· πŸ“¦ 3 Β· πŸ“‹ 16 - 12% open Β· ⏱️ 27.03.2024): ``` git clone https://github.com/ur-whitelab/chemcrow-public ``` - [PyPi](https://pypi.org/project/chemcrow) (πŸ“₯ 1K / month): ``` pip install chemcrow ```
ChemNLP project (πŸ₯ˆ14 Β· ⭐ 140) - ChemNLP project. MIT datasets - [GitHub](https://github.com/OpenBioML/chemnlp) (πŸ‘¨β€πŸ’» 26 Β· πŸ”€ 46 Β· πŸ“‹ 250 - 44% open Β· ⏱️ 01.04.2024): ``` git clone https://github.com/OpenBioML/chemnlp ``` - [PyPi](https://pypi.org/project/chemnlp) (πŸ“₯ 39 / month): ``` pip install chemnlp ```
gptchem (πŸ₯ˆ13 Β· ⭐ 220 Β· πŸ’€) - Use GPT-3 to solve chemistry problems. MIT - [GitHub](https://github.com/kjappelbaum/gptchem) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 39 Β· πŸ“‹ 21 - 90% open Β· ⏱️ 04.10.2023): ``` git clone https://github.com/kjappelbaum/gptchem ``` - [PyPi](https://pypi.org/project/gptchem) (πŸ“₯ 44 / month): ``` pip install gptchem ```
LLaMP (πŸ₯ˆ12 Β· ⭐ 43) - A web app and Python API for multi-modal RAG framework to ground LLMs on high-fidelity materials informatics. An.. BSD-3 materials-discovery cheminformatics generative MD language-models Python general-tool - [GitHub](https://github.com/chiang-yuan/llamp) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 3 Β· πŸ“‹ 25 - 32% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/chiang-yuan/llamp ```
MoLFormer (πŸ₯‰9 Β· ⭐ 220 Β· πŸ’€) - Repository for MolFormer. Apache-2 transformer pre-trained drug-discovery - [GitHub](https://github.com/IBM/molformer) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 39 Β· πŸ“‹ 18 - 44% open Β· ⏱️ 16.10.2023): ``` git clone https://github.com/IBM/molformer ```
MolSkill (πŸ₯‰9 Β· ⭐ 99 Β· πŸ’€) - Extracting medicinal chemistry intuition via preference machine learning. MIT drug-discovery recommender - [GitHub](https://github.com/microsoft/molskill) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 8 Β· πŸ“‹ 5 - 40% open Β· ⏱️ 31.10.2023): ``` git clone https://github.com/microsoft/molskill ``` - [Conda](https://anaconda.org/msr-ai4science/molskill) (πŸ“₯ 260 Β· ⏱️ 18.06.2023): ``` conda install -c msr-ai4science molskill ```
chemlift (πŸ₯‰7 Β· ⭐ 30 Β· πŸ’€) - Language-interfaced fine-tuning for chemistry. MIT - [GitHub](https://github.com/lamalab-org/chemlift) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 3 Β· πŸ“‹ 18 - 61% open Β· ⏱️ 14.10.2023): ``` git clone https://github.com/lamalab-org/chemlift ```
LLM-Prop (πŸ₯‰7 Β· ⭐ 24) - A repository for the LLM-Prop implementation. MIT - [GitHub](https://github.com/vertaix/LLM-Prop) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 5 Β· ⏱️ 26.04.2024): ``` git clone https://github.com/vertaix/LLM-Prop ```
SciBot (πŸ₯‰5 Β· ⭐ 28) - SciBot is a simple demo of building a domain-specific chatbot for science. Unlicensed - [GitHub](https://github.com/CFN-softbio/SciBot) (πŸ”€ 8 Β· πŸ“¦ 1 Β· ⏱️ 19.04.2024): ``` git clone https://github.com/CFN-softbio/SciBot ```
BERT-PSIE-TC (πŸ₯‰5 Β· ⭐ 10 Β· πŸ’€) - A dataset of Curie temperatures automatically extracted from scientific literature with the use of the BERT-PSIE.. MIT magnetism - [GitHub](https://github.com/StefanoSanvitoGroup/BERT-PSIE-TC) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 3 Β· ⏱️ 18.08.2023): ``` git clone https://github.com/StefanoSanvitoGroup/BERT-PSIE-TC ```
MAPI_LLM (πŸ₯‰5 Β· ⭐ 8) - A LLM application developed during the LLM March MADNESS Hackathon https://doi.org/10.1039/D3DD00113J. MIT dataset - [GitHub](https://github.com/maykcaldas/MAPI_LLM) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 2 Β· ⏱️ 11.04.2024): ``` git clone https://github.com/maykcaldas/MAPI_LLM ```
Show 5 hidden projects... - ChemDataExtractor (πŸ₯‡16 Β· ⭐ 290 Β· πŸ’€) - Automatically extract chemical information from scientific documents. MIT literature-data - mat2vec (πŸ₯ˆ12 Β· ⭐ 610 Β· πŸ’€) - Supplementary Materials for Tshitoyan et al. Unsupervised word embeddings capture latent knowledge from materials.. MIT rep-learn - nlcc (πŸ₯‰11 Β· ⭐ 44 Β· πŸ’€) - Natural language computational chemistry command line interface. MIT single-paper - ChemDataWriter (πŸ₯‰4 Β· ⭐ 14 Β· πŸ’€) - ChemDataWriter is a transformer-based library for automatically generating research books in the chemistry area. MIT literature-data - CatBERTa (πŸ₯‰3 Β· ⭐ 18) - Large Language Model for Catalyst Property Prediction. Unlicensed transformer catalysis


Materials Discovery

Back to top

Projects that implement materials discovery methods using atomistic ML.

aviary (πŸ₯‡12 Β· ⭐ 44) - The Wren sits on its Roost in the Aviary. MIT - [GitHub](https://github.com/CompRhys/aviary) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 10 Β· πŸ“‹ 26 - 7% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/CompRhys/aviary ```
Materials Discovery: GNoME (πŸ₯ˆ10 Β· ⭐ 840 Β· πŸ’€) - Graph Networks for Materials Science (GNoME) and dataset of 381,000 novel stable materials. Apache-2 rep-learn datasets - [GitHub](https://github.com/google-deepmind/materials_discovery) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 130 Β· πŸ“‹ 20 - 80% open Β· ⏱️ 02.12.2023): ``` git clone https://github.com/google-deepmind/materials_discovery ```
Show 7 hidden projects... - BOSS (πŸ₯ˆ7 Β· ⭐ 20 Β· πŸ’€) - Bayesian Optimization Structure Search (BOSS). Unlicensed probabilistic - AGOX (πŸ₯ˆ6 Β· ⭐ 13 Β· πŸ’€) - AGOX is a package for global optimization of atomic system using e.g. the energy calculated from density functional.. GPL-3.0 structure-optimization - Computational Autonomy for Materials Discovery (CAMD) (πŸ₯ˆ6 Β· ⭐ 1 Β· πŸ’€) - Agent-based sequential learning software for materials discovery. Apache-2 - closed-loop-acceleration-benchmarks (πŸ₯‰4 Β· πŸ’€) - Data and scripts in support of the publication By how much can closed-loop frameworks accelerate computational.. MIT materials-discovery active-learning single-paper - SPINNER (πŸ₯‰3 Β· ⭐ 11 Β· πŸ’€) - SPINNER (Structure Prediction of Inorganic crystals using Neural Network potentials with Evolutionary and Random.. GPL-3.0 C++ structure-prediction - sl_discovery (πŸ₯‰3 Β· ⭐ 5 Β· πŸ’€) - Data processing and models related to Quantifying the performance of machine learning models in materials discovery. Apache-2 materials-discovery single-paper - CSPML (crystal structure prediction with machine learning-based element substitution) (πŸ₯‰2 Β· ⭐ 16 Β· πŸ’€) - Original implementation of CSPML. Unlicensed structure-prediction


Mathematical tools

Back to top

Projects that implement mathematical objects used in atomistic machine learning.

KFAC-JAX (πŸ₯‡20 Β· ⭐ 210) - Second Order Optimization and Curvature Estimation with K-FAC in JAX. Apache-2 - [GitHub](https://github.com/google-deepmind/kfac-jax) (πŸ‘¨β€πŸ’» 13 Β· πŸ”€ 16 Β· πŸ“¦ 10 Β· πŸ“‹ 12 - 25% open Β· ⏱️ 24.06.2024): ``` git clone https://github.com/deepmind/kfac-jax ``` - [PyPi](https://pypi.org/project/kfac-jax) (πŸ“₯ 1.3K / month): ``` pip install kfac-jax ```
gpax (πŸ₯‡17 Β· ⭐ 190 Β· πŸ“‰) - Gaussian Processes for Experimental Sciences. MIT probabilistic active-learning - [GitHub](https://github.com/ziatdinovmax/gpax) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 23 Β· πŸ“¦ 1 Β· πŸ“‹ 39 - 17% open Β· ⏱️ 21.05.2024): ``` git clone https://github.com/ziatdinovmax/gpax ``` - [PyPi](https://pypi.org/project/gpax) (πŸ“₯ 260 / month): ``` pip install gpax ```
Polynomials4ML.jl (πŸ₯ˆ15 Β· ⭐ 12) - Polynomials for ML: fast evaluation, batching, differentiation. MIT Julia - [GitHub](https://github.com/ACEsuit/Polynomials4ML.jl) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 5 Β· πŸ“‹ 51 - 33% open Β· ⏱️ 22.06.2024): ``` git clone https://github.com/ACEsuit/Polynomials4ML.jl ```
SpheriCart (πŸ₯ˆ14 Β· ⭐ 58) - Multi-language library for the calculation of spherical harmonics in Cartesian coordinates. MIT - [GitHub](https://github.com/lab-cosmo/sphericart) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 10 Β· πŸ“₯ 59 Β· πŸ“¦ 2 Β· πŸ“‹ 27 - 62% open Β· ⏱️ 12.06.2024): ``` git clone https://github.com/lab-cosmo/sphericart ``` - [PyPi](https://pypi.org/project/sphericart) (πŸ“₯ 130 / month): ``` pip install sphericart ```
GElib (πŸ₯ˆ9 Β· ⭐ 19) - C++/CUDA library for SO(3) equivariant operations. MPL-2.0 C++ - [GitHub](https://github.com/risi-kondor/GElib) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 3 Β· πŸ“‹ 5 - 40% open Β· ⏱️ 26.04.2024): ``` git clone https://github.com/risi-kondor/GElib ```
COSMO Toolbox (πŸ₯‰6 Β· ⭐ 6) - Assorted libraries and utilities for atomistic simulation analysis. Unlicensed C++ - [GitHub](https://github.com/lab-cosmo/toolbox) (πŸ‘¨β€πŸ’» 9 Β· πŸ”€ 5 Β· ⏱️ 19.03.2024): ``` git clone https://github.com/lab-cosmo/toolbox ```
Show 5 hidden projects... - lie-nn (πŸ₯ˆ9 Β· ⭐ 26 Β· πŸ’€) - Tools for building equivariant polynomials on reductive Lie groups. MIT rep-learn - cnine (πŸ₯‰6 Β· ⭐ 4) - Cnine tensor library. Unlicensed C++ - EquivariantOperators.jl (πŸ₯‰5 Β· ⭐ 18 Β· πŸ’€) - MIT Julia - torch_spex (πŸ₯‰3 Β· ⭐ 3) - Spherical expansions in PyTorch. Unlicensed - Wigner Kernels (πŸ₯‰2 Β· ⭐ 1 Β· πŸ’€) - Collection of programs to benchmark Wigner kernels. Unlicensed benchmarking


Molecular Dynamics

Back to top

Projects that simplify the integration of molecular dynamics and atomistic machine learning.

JAX-MD (πŸ₯‡24 Β· ⭐ 1.1K) - Differentiable, Hardware Accelerated, Molecular Dynamics. Apache-2 - [GitHub](https://github.com/jax-md/jax-md) (πŸ‘¨β€πŸ’» 31 Β· πŸ”€ 180 Β· πŸ“¦ 52 Β· πŸ“‹ 150 - 45% open Β· ⏱️ 17.04.2024): ``` git clone https://github.com/jax-md/jax-md ``` - [PyPi](https://pypi.org/project/jax-md) (πŸ“₯ 2.6K / month): ``` pip install jax-md ```
mlcolvar (πŸ₯ˆ19 Β· ⭐ 86) - A unified framework for machine learning collective variables for enhanced sampling simulations. MIT enhanced-sampling - [GitHub](https://github.com/luigibonati/mlcolvar) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 21 Β· πŸ“¦ 1 Β· πŸ“‹ 68 - 16% open Β· ⏱️ 14.06.2024): ``` git clone https://github.com/luigibonati/mlcolvar ``` - [PyPi](https://pypi.org/project/mlcolvar) (πŸ“₯ 320 / month): ``` pip install mlcolvar ```
FitSNAP (πŸ₯ˆ17 Β· ⭐ 140) - Software for generating SNAP machine-learning interatomic potentials. GPL-2.0 - [GitHub](https://github.com/FitSNAP/FitSNAP) (πŸ‘¨β€πŸ’» 24 Β· πŸ”€ 46 Β· πŸ“₯ 8 Β· πŸ“¦ 2 Β· πŸ“‹ 65 - 13% open Β· ⏱️ 27.06.2024): ``` git clone https://github.com/FitSNAP/FitSNAP ``` - [Conda](https://anaconda.org/conda-forge/fitsnap3) (πŸ“₯ 6.9K Β· ⏱️ 16.06.2023): ``` conda install -c conda-forge fitsnap3 ```
openmm-torch (πŸ₯ˆ16 Β· ⭐ 170) - OpenMM plugin to define forces with neural networks. Custom ML-IAP C++ - [GitHub](https://github.com/openmm/openmm-torch) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 24 Β· πŸ“‹ 87 - 26% open Β· ⏱️ 27.05.2024): ``` git clone https://github.com/openmm/openmm-torch ``` - [Conda](https://anaconda.org/conda-forge/openmm-torch) (πŸ“₯ 340K Β· ⏱️ 03.06.2024): ``` conda install -c conda-forge openmm-torch ```
OpenMM-ML (πŸ₯‰14 Β· ⭐ 75) - High level API for using machine learning models in OpenMM simulations. MIT ML-IAP - [GitHub](https://github.com/openmm/openmm-ml) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 19 Β· πŸ“‹ 51 - 39% open Β· ⏱️ 06.06.2024): ``` git clone https://github.com/openmm/openmm-ml ``` - [Conda](https://anaconda.org/conda-forge/openmm-ml) (πŸ“₯ 4K Β· ⏱️ 07.06.2024): ``` conda install -c conda-forge openmm-ml ```
pair_nequip (πŸ₯‰11 Β· ⭐ 37) - LAMMPS pair style for NequIP. MIT ML-IAP rep-learn - [GitHub](https://github.com/mir-group/pair_nequip) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 13 Β· πŸ“‹ 28 - 28% open Β· ⏱️ 05.06.2024): ``` git clone https://github.com/mir-group/pair_nequip ```
pair_allegro (πŸ₯‰9 Β· ⭐ 33) - LAMMPS pair style for Allegro deep learning interatomic potentials with parallelization support. MIT ML-IAP rep-learn - [GitHub](https://github.com/mir-group/pair_allegro) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 7 Β· πŸ“‹ 26 - 34% open Β· ⏱️ 05.06.2024): ``` git clone https://github.com/mir-group/pair_allegro ```
PACE (πŸ₯‰9 Β· ⭐ 21 Β· πŸ’€) - The LAMMPS ML-IAP `pair_style pace`, aka Atomic Cluster Expansion (ACE), aka ML-PACE,.. Custom - [GitHub](https://github.com/ICAMS/lammps-user-pace) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 10 Β· ⏱️ 27.11.2023): ``` git clone https://github.com/ICAMS/lammps-user-pace ```
SOMD (πŸ₯‰6 Β· ⭐ 11) - Molecular dynamics package designed for the SIESTA DFT code. AGPL-3.0 ML-IAP active-learning - [GitHub](https://github.com/initqp/somd) (πŸ”€ 2 Β· ⏱️ 03.07.2024): ``` git clone https://github.com/initqp/somd ```
Show 1 hidden projects... - interface-lammps-mlip-3 (πŸ₯‰3 Β· ⭐ 5 Β· πŸ’€) - An interface between LAMMPS and MLIP (version 3). GPL-2.0


Reinforcement Learning

Back to top

Projects that focus on reinforcement learning for atomistic ML.

Show 2 hidden projects... - ReLeaSE (πŸ₯‡11 Β· ⭐ 340 Β· πŸ’€) - Deep Reinforcement Learning for de-novo Drug Design. MIT drug-discovery - CatGym (πŸ₯‰6 Β· ⭐ 11 Β· πŸ’€) - Surface segregation using Deep Reinforcement Learning. GPL


Representation Engineering

Back to top

Projects that offer implementations of representations aka descriptors, fingerprints of atomistic systems, and models built with them, aka feature engineering.

cdk (πŸ₯‡23 Β· ⭐ 480) - The Chemistry Development Kit. LGPL-2.1 cheminformatics Java - [GitHub](https://github.com/cdk/cdk) (πŸ‘¨β€πŸ’» 160 Β· πŸ”€ 150 Β· πŸ“₯ 21K Β· πŸ“‹ 280 - 10% open Β· ⏱️ 28.06.2024): ``` git clone https://github.com/cdk/cdk ``` - [Maven](https://search.maven.org/artifact/org.openscience.cdk/cdk-bundle): ``` org.openscience.cdk cdk-bundle [VERSION] ```
DScribe (πŸ₯‡22 Β· ⭐ 380) - DScribe is a python package for creating machine learning descriptors for atomistic systems. Apache-2 - [GitHub](https://github.com/SINGROUP/dscribe) (πŸ‘¨β€πŸ’» 18 Β· πŸ”€ 85 Β· πŸ“¦ 190 Β· πŸ“‹ 98 - 7% open Β· ⏱️ 28.05.2024): ``` git clone https://github.com/SINGROUP/dscribe ``` - [PyPi](https://pypi.org/project/dscribe) (πŸ“₯ 300K / month): ``` pip install dscribe ``` - [Conda](https://anaconda.org/conda-forge/dscribe) (πŸ“₯ 110K Β· ⏱️ 28.05.2024): ``` conda install -c conda-forge dscribe ```
MODNet (πŸ₯‡16 Β· ⭐ 71) - MODNet: a framework for machine learning materials properties. MIT pre-trained small-data transfer-learning - [GitHub](https://github.com/ppdebreuck/modnet) (πŸ‘¨β€πŸ’» 8 Β· πŸ”€ 31 Β· πŸ“¦ 6 Β· πŸ“‹ 42 - 38% open Β· ⏱️ 24.06.2024): ``` git clone https://github.com/ppdebreuck/modnet ```
GlassPy (πŸ₯ˆ14 Β· ⭐ 26) - Python module for scientists working with glass materials. GPL-3.0 - [GitHub](https://github.com/drcassar/glasspy) (πŸ”€ 7 Β· πŸ“¦ 5 Β· πŸ“‹ 6 - 16% open Β· ⏱️ 21.01.2024): ``` git clone https://github.com/drcassar/glasspy ``` - [PyPi](https://pypi.org/project/glasspy) (πŸ“₯ 330 / month): ``` pip install glasspy ```
SISSO (πŸ₯ˆ13 Β· ⭐ 220 Β· πŸ’€) - A data-driven method combining symbolic regression and compressed sensing for accurate & interpretable models. Apache-2 Fortran - [GitHub](https://github.com/rouyang2017/SISSO) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 74 Β· πŸ“‹ 61 - 6% open Β· ⏱️ 12.09.2023): ``` git clone https://github.com/rouyang2017/SISSO ```
Librascal (πŸ₯ˆ13 Β· ⭐ 79 Β· πŸ’€) - A scalable and versatile library to generate representations for atomic-scale learning. LGPL-2.1 - [GitHub](https://github.com/lab-cosmo/librascal) (πŸ‘¨β€πŸ’» 30 Β· πŸ”€ 19 Β· πŸ“‹ 230 - 43% open Β· ⏱️ 30.11.2023): ``` git clone https://github.com/lab-cosmo/librascal ```
Rascaline (πŸ₯ˆ12 Β· ⭐ 44) - Computing representations for atomistic machine learning. BSD-3 Rust C++ - [GitHub](https://github.com/Luthaf/rascaline) (πŸ‘¨β€πŸ’» 14 Β· πŸ”€ 13 Β· πŸ“‹ 65 - 53% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/Luthaf/rascaline ```
NICE (πŸ₯‰6 Β· ⭐ 12) - NICE (N-body Iteratively Contracted Equivariants) is a set of tools designed for the calculation of invariant and.. MIT - [GitHub](https://github.com/lab-cosmo/nice) (πŸ‘¨β€πŸ’» 4 Β· πŸ”€ 3 Β· πŸ“‹ 3 - 66% open Β· ⏱️ 15.04.2024): ``` git clone https://github.com/lab-cosmo/nice ```
Show 15 hidden projects... - CatLearn (πŸ₯‡16 Β· ⭐ 98 Β· πŸ’€) - GPL-3.0 surface-science - cmlkit (πŸ₯ˆ10 Β· ⭐ 34 Β· πŸ’€) - tools for machine learning in condensed matter physics and quantum chemistry. MIT benchmarking - BenchML (πŸ₯‰9 Β· ⭐ 15 Β· πŸ’€) - ML benchmarking and pipeling framework. Apache-2 benchmarking - CBFV (πŸ₯‰8 Β· ⭐ 21 Β· πŸ’€) - Tool to quickly create a composition-based feature vector. Unlicensed - SkipAtom (πŸ₯‰7 Β· ⭐ 23 Β· πŸ’€) - Distributed representations of atoms, inspired by the Skip-gram model. MIT - milad (πŸ₯‰6 Β· ⭐ 29 Β· πŸ’€) - Moment Invariants Local Atomic Descriptor. GPL-3.0 generative - SA-GPR (πŸ₯‰6 Β· ⭐ 18 Β· πŸ’€) - Public repository for symmetry-adapted Gaussian Process Regression (SA-GPR). LGPL-3.0 C-lang - fplib (πŸ₯‰6 Β· ⭐ 7 Β· πŸ’€) - a fingerprint library. MIT C-lang single-paper - SOAPxx (πŸ₯‰6 Β· ⭐ 7 Β· πŸ’€) - A SOAP implementation. GPL-2.0 C++ - soap_turbo (πŸ₯‰6 Β· ⭐ 4 Β· πŸ’€) - soap_turbo comprises a series of libraries to be used in combination with QUIP/GAP and TurboGAP. Custom Fortran - pyLODE (πŸ₯‰6 Β· ⭐ 2 Β· πŸ’€) - Pythonic implementation of LOng Distance Equivariants. Apache-2 electrostatics - SISSO++ (πŸ₯‰5 Β· ⭐ 3 Β· πŸ’€) - C++ Implementation of SISSO with python bindings. Apache-2 C++ - magnetism-prediction (πŸ₯‰4 Β· ⭐ 1 Β· πŸ’€) - DFT-aided Machine Learning Search for Magnetism in Fe-based Bimetallic Chalcogenides. Apache-2 magnetism single-paper - ML-for-CurieTemp-Predictions (πŸ₯‰3 Β· ⭐ 1 Β· πŸ’€) - Machine Learning Predictions of High-Curie-Temperature Materials. MIT single-paper magnetism - AMP (πŸ₯‰2) - Amp is an open-source package designed to easily bring machine-learning to atomistic calculations. Unlicensed


Representation Learning

Back to top

General models that learn a representations aka embeddings of atomistic systems, such as message-passing neural networks (MPNN).

Deep Graph Library (DGL) (πŸ₯‡38 Β· ⭐ 13K) - Python package built to ease deep learning on graph, on top of existing DL frameworks. Apache-2 - [GitHub](https://github.com/dmlc/dgl) (πŸ‘¨β€πŸ’» 290 Β· πŸ”€ 2.9K Β· πŸ“¦ 260 Β· πŸ“‹ 2.7K - 14% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/dmlc/dgl ``` - [PyPi](https://pypi.org/project/dgl) (πŸ“₯ 140K / month): ``` pip install dgl ``` - [Conda](https://anaconda.org/dglteam/dgl) (πŸ“₯ 350K Β· ⏱️ 28.06.2024): ``` conda install -c dglteam dgl ```
PyG Models (πŸ₯‡35 Β· ⭐ 21K) - Representation learning models implemented in PyTorch Geometric. MIT general-ml - [GitHub](https://github.com/pyg-team/pytorch_geometric) (πŸ‘¨β€πŸ’» 500 Β· πŸ”€ 3.5K Β· πŸ“¦ 6K Β· πŸ“‹ 3.5K - 24% open Β· ⏱️ 03.07.2024): ``` git clone https://github.com/pyg-team/pytorch_geometric ```
SchNetPack (πŸ₯‡26 Β· ⭐ 750 Β· πŸ“ˆ) - SchNetPack - Deep Neural Networks for Atomistic Systems. MIT - [GitHub](https://github.com/atomistic-machine-learning/schnetpack) (πŸ‘¨β€πŸ’» 35 Β· πŸ”€ 200 Β· πŸ“¦ 82 Β· πŸ“‹ 240 - 1% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/atomistic-machine-learning/schnetpack ``` - [PyPi](https://pypi.org/project/schnetpack) (πŸ“₯ 950 / month): ``` pip install schnetpack ```
e3nn (πŸ₯‡25 Β· ⭐ 890) - A modular framework for neural networks with Euclidean symmetry. MIT - [GitHub](https://github.com/e3nn/e3nn) (πŸ‘¨β€πŸ’» 31 Β· πŸ”€ 130 Β· πŸ“¦ 260 Β· πŸ“‹ 150 - 11% open Β· ⏱️ 23.06.2024): ``` git clone https://github.com/e3nn/e3nn ``` - [PyPi](https://pypi.org/project/e3nn) (πŸ“₯ 400K / month): ``` pip install e3nn ``` - [Conda](https://anaconda.org/conda-forge/e3nn) (πŸ“₯ 18K Β· ⏱️ 18.06.2023): ``` conda install -c conda-forge e3nn ```
MatGL (Materials Graph Library) (πŸ₯‡22 Β· ⭐ 230) - Graph deep learning library for materials. BSD-3 - [GitHub](https://github.com/materialsvirtuallab/matgl) (πŸ‘¨β€πŸ’» 17 Β· πŸ”€ 57 Β· πŸ“¦ 40 Β· πŸ“‹ 78 - 2% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/materialsvirtuallab/matgl ``` - [PyPi](https://pypi.org/project/m3gnet) (πŸ“₯ 770 / month): ``` pip install m3gnet ```
NVIDIA Deep Learning Examples for Tensor Cores (πŸ₯ˆ21 Β· ⭐ 13K) - State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and.. Custom educational drug-discovery - [GitHub](https://github.com/NVIDIA/DeepLearningExamples) (πŸ‘¨β€πŸ’» 120 Β· πŸ”€ 3K Β· πŸ“‹ 820 - 30% open Β· ⏱️ 04.04.2024): ``` git clone https://github.com/NVIDIA/DeepLearningExamples ```
DIG: Dive into Graphs (πŸ₯ˆ20 Β· ⭐ 1.8K Β· πŸ“‰) - A library for graph deep learning research. GPL-3.0 - [GitHub](https://github.com/divelab/DIG) (πŸ‘¨β€πŸ’» 50 Β· πŸ”€ 270 Β· πŸ“‹ 200 - 13% open Β· ⏱️ 04.02.2024): ``` git clone https://github.com/divelab/DIG ``` - [PyPi](https://pypi.org/project/dive-into-graphs) (πŸ“₯ 480 / month): ``` pip install dive-into-graphs ```
ALIGNN (πŸ₯ˆ20 Β· ⭐ 190) - Atomistic Line Graph Neural Network https://scholar.google.com/citations?user=9Q-tNnwAAAAJ&hl=en. Custom - [GitHub](https://github.com/usnistgov/alignn) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 77 Β· πŸ“¦ 12 Β· πŸ“‹ 54 - 55% open Β· ⏱️ 28.06.2024): ``` git clone https://github.com/usnistgov/alignn ``` - [PyPi](https://pypi.org/project/alignn) (πŸ“₯ 720 / month): ``` pip install alignn ```
kgcnn (πŸ₯ˆ20 Β· ⭐ 100) - Graph convolutions in Keras with TensorFlow, PyTorch or Jax. MIT - [GitHub](https://github.com/aimat-lab/gcnn_keras) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 28 Β· πŸ“¦ 18 Β· πŸ“‹ 86 - 13% open Β· ⏱️ 06.05.2024): ``` git clone https://github.com/aimat-lab/gcnn_keras ``` - [PyPi](https://pypi.org/project/kgcnn) (πŸ“₯ 280 / month): ``` pip install kgcnn ```
e3nn-jax (πŸ₯ˆ17 Β· ⭐ 170) - jax library for E3 Equivariant Neural Networks. Apache-2 - [GitHub](https://github.com/e3nn/e3nn-jax) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 18 Β· πŸ“¦ 36 Β· ⏱️ 24.06.2024): ``` git clone https://github.com/e3nn/e3nn-jax ``` - [PyPi](https://pypi.org/project/e3nn-jax) (πŸ“₯ 1.9K / month): ``` pip install e3nn-jax ```
matsciml (πŸ₯ˆ17 Β· ⭐ 130) - Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery.. MIT workflows benchmarking - [GitHub](https://github.com/IntelLabs/matsciml) (πŸ‘¨β€πŸ’» 11 Β· πŸ”€ 17 Β· πŸ“‹ 49 - 32% open Β· ⏱️ 01.07.2024): ``` git clone https://github.com/IntelLabs/matsciml ```
Uni-Mol (πŸ₯ˆ16 Β· ⭐ 600) - Official Repository for the Uni-Mol Series Methods. MIT pre-trained - [GitHub](https://github.com/deepmodeling/Uni-Mol) (πŸ‘¨β€πŸ’» 16 Β· πŸ”€ 110 Β· πŸ“₯ 14K Β· πŸ“‹ 140 - 39% open Β· ⏱️ 04.07.2024): ``` git clone https://github.com/dptech-corp/Uni-Mol ```
escnn (πŸ₯ˆ14 Β· ⭐ 330 Β· πŸ’€) - Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/. Custom - [GitHub](https://github.com/QUVA-Lab/escnn) (πŸ‘¨β€πŸ’» 10 Β· πŸ”€ 42 Β· πŸ“‹ 62 - 41% open Β· ⏱️ 17.10.2023): ``` git clone https://github.com/QUVA-Lab/escnn ``` - [PyPi](https://pypi.org/project/escnn) (πŸ“₯ 260 / month): ``` pip install escnn ```
hippynn (πŸ₯ˆ11 Β· ⭐ 59) - python library for atomistic machine learning. Custom workflows - [GitHub](https://github.com/lanl/hippynn) (πŸ‘¨β€πŸ’» 13 Β· πŸ”€ 22 Β· πŸ“¦ 1 Β· πŸ“‹ 14 - 42% open Β· ⏱️ 24.06.2024): ``` git clone https://github.com/lanl/hippynn ```
EquiformerV2 (πŸ₯ˆ9 Β· ⭐ 160) - [ICLR24] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations. MIT - [GitHub](https://github.com/atomicarchitects/equiformer_v2) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 22 Β· ⏱️ 08.05.2024): ``` git clone https://github.com/atomicarchitects/equiformer_v2 ```
graphite (πŸ₯ˆ9 Β· ⭐ 51) - A repository for implementing graph network models based on atomic structures. MIT - [GitHub](https://github.com/LLNL/graphite) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 9 Β· πŸ“¦ 11 Β· πŸ“‹ 3 - 66% open Β· ⏱️ 13.05.2024): ``` git clone https://github.com/llnl/graphite ```
ai4material_design (πŸ₯ˆ9 Β· ⭐ 5 Β· πŸ’€) - Code for Kazeev, N., Al-Maeeni, A.R., Romanov, I. et al. Sparse representation for machine learning the properties of.. Apache-2 pre-trained material-defect - [GitHub](https://github.com/HSE-LAMBDA/ai4material_design) (πŸ‘¨β€πŸ’» 11 Β· πŸ”€ 3 Β· ⏱️ 21.11.2023): ``` git clone https://github.com/HSE-LAMBDA/ai4material_design ```
DeeperGATGNN (πŸ₯‰8 Β· ⭐ 43) - Scalable graph neural networks for materials property prediction. MIT - [GitHub](https://github.com/usccolumbia/deeperGATGNN) (πŸ‘¨β€πŸ’» 3 Β· πŸ”€ 8 Β· ⏱️ 19.01.2024): ``` git clone https://github.com/usccolumbia/deeperGATGNN ```
AdsorbML (πŸ₯‰8 Β· ⭐ 33 Β· πŸ’€) - MIT surface-science single-paper - [GitHub](https://github.com/Open-Catalyst-Project/AdsorbML) (πŸ‘¨β€πŸ’» 5 Β· πŸ”€ 4 Β· πŸ“‹ 4 - 75% open Β· ⏱️ 31.07.2023): ``` git clone https://github.com/Open-Catalyst-Project/AdsorbML ```
ML4pXRDs (πŸ₯‰7 Β· πŸ’€) - Contains code to train neural networks based on simulated powder XRDs from synthetic crystals. MIT XRD single-paper - [GitHub](https://github.com/aimat-lab/ML4pXRDs) (πŸ“₯ 2 Β· ⏱️ 14.07.2023): ``` git clone https://github.com/aimat-lab/ML4pXRDs ```
Show 34 hidden projects... - dgl-lifesci (πŸ₯‡23 Β· ⭐ 700 Β· πŸ’€) - Python package for graph neural networks in chemistry and biology. Apache-2 - benchmarking-gnns (πŸ₯ˆ14 Β· ⭐ 2.5K Β· πŸ’€) - Repository for benchmarking graph neural networks. MIT single-paper benchmarking - Crystal Graph Convolutional Neural Networks (CGCNN) (πŸ₯ˆ12 Β· ⭐ 610 Β· πŸ’€) - Crystal graph convolutional neural networks for predicting material properties. MIT - Neural fingerprint (nfp) (πŸ₯ˆ12 Β· ⭐ 57 Β· πŸ’€) - Keras layers for end-to-end learning with rdkit and pymatgen. Custom - Compositionally-Restricted Attention-Based Network (CrabNet) (πŸ₯ˆ11 Β· ⭐ 12 Β· πŸ’€) - Predict materials properties using only the composition information!. MIT - GDC (πŸ₯ˆ10 Β· ⭐ 260 Β· πŸ’€) - Graph Diffusion Convolution, as proposed in Diffusion Improves Graph Learning (NeurIPS 2019). MIT generative - SE(3)-Transformers (πŸ₯ˆ9 Β· ⭐ 480 Β· πŸ’€) - code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503. MIT single-paper transformer - molecularGNN_smiles (πŸ₯ˆ9 Β· ⭐ 280 Β· πŸ’€) - The code of a graph neural network (GNN) for molecules, which is based on learning representations of r-radius.. Apache-2 - GATGNN: Global Attention Graph Neural Network (πŸ₯ˆ9 Β· ⭐ 66 Β· πŸ’€) - Pytorch Repository for our work: Graph convolutional neural networks with global attention for improved materials.. MIT - FAENet (πŸ₯ˆ9 Β· ⭐ 25 Β· πŸ’€) - MIT - Equiformer (πŸ₯‰8 Β· ⭐ 180 Β· πŸ’€) - [ICLR23 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs. MIT transformer - CGAT (πŸ₯‰8 Β· ⭐ 25 Β· πŸ’€) - Crystal graph attention neural networks for materials prediction. MIT - UVVisML (πŸ₯‰8 Β· ⭐ 19 Β· πŸ’€) - Predict optical properties of molecules with machine learning. MIT optical-properties single-paper probabilistic - T-e3nn (πŸ₯‰8 Β· ⭐ 8 Β· πŸ’€) - Time-reversal Euclidean neural networks based on e3nn. MIT magnetism - DTNN (πŸ₯‰7 Β· ⭐ 76 Β· πŸ’€) - Deep Tensor Neural Network. MIT - Cormorant (πŸ₯‰7 Β· ⭐ 60 Β· πŸ’€) - Codebase for Cormorant Neural Networks. Custom - escnn_jax (πŸ₯‰7 Β· ⭐ 25 Β· πŸ’€) - Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/. Custom - tensorfieldnetworks (πŸ₯‰6 Β· ⭐ 150 Β· πŸ’€) - MIT - MACE-Layer (πŸ₯‰6 Β· ⭐ 32 Β· πŸ’€) - Higher order equivariant graph neural networks for 3D point clouds. MIT - charge_transfer_nnp (πŸ₯‰6 Β· ⭐ 27 Β· πŸ’€) - Graph neural network potential with charge transfer. MIT electrostatics - GLAMOUR (πŸ₯‰6 Β· ⭐ 18 Β· πŸ’€) - Graph Learning over Macromolecule Representations. MIT single-paper - Autobahn (πŸ₯‰5 Β· ⭐ 30 Β· πŸ’€) - Repository for Autobahn: Automorphism Based Graph Neural Networks. MIT - SCFNN (πŸ₯‰5 Β· ⭐ 15 Β· πŸ’€) - Self-consistent determination of long-range electrostatics in neural network potentials. MIT C++ electrostatics single-paper - CraTENet (πŸ₯‰5 Β· ⭐ 12 Β· πŸ’€) - An attention-based deep neural network for thermoelectric transport properties. MIT transport-phenomena - Per-Site CGCNN (πŸ₯‰5 Β· ⭐ 1 Β· πŸ’€) - Crystal graph convolutional neural networks for predicting material properties. MIT pre-trained single-paper - Per-site PAiNN (πŸ₯‰5 Β· ⭐ 1 Β· πŸ’€) - Fork of PaiNN for PerovskiteOrderingGCNNs. MIT probabilistic pre-trained single-paper - FieldSchNet (πŸ₯‰4 Β· ⭐ 16 Β· πŸ’€) - MIT - Graph Transport Network (πŸ₯‰4 Β· ⭐ 16 Β· πŸ’€) - Graph transport network (GTN), as proposed in Scalable Optimal Transport in High Dimensions for Graph Distances,.. Custom transport-phenomena - gkx: Green-Kubo Method in JAX (πŸ₯‰4 Β· ⭐ 3) - Green-Kubo + JAX + MLPs = Anharmonic Thermal Conductivities Done Fast. MIT transport-phenomena - Atom2Vec (πŸ₯‰3 Β· ⭐ 31) - Atom2Vec: a simple way to describe atoms for machine learning. Unlicensed - atom_by_atom (πŸ₯‰3 Β· ⭐ 6 Β· πŸ’€) - Atom-by-atom design of metal oxide catalysts for the oxygen evolution reaction with Machine Learning. Unlicensed surface-science single-paper - Element encoder (πŸ₯‰3 Β· ⭐ 6 Β· πŸ’€) - Autoencoder neural network to compress properties of atomic species into a vector representation. GPL-3.0 single-paper - Point Edge Transformer (πŸ₯‰2) - Smooth, exact rotational symmetrization for deep learning on point clouds. CC-BY-4.0 - SphericalNet (πŸ₯‰1 Β· ⭐ 3 Β· πŸ’€) - Implementation of Clebsch-Gordan Networks (CGnet: https://arxiv.org/pdf/1806.09231.pdf) by GElib & cnine libraries in.. Unlicensed


Unsupervised Learning

Back to top

Projects that focus on unsupervised learning (USL) for atomistic ML, such as dimensionality reduction, clustering and visualization.

DADApy (πŸ₯‡18 Β· ⭐ 97) - Distance-based Analysis of DAta-manifolds in python. Apache-2 - [GitHub](https://github.com/sissa-data-science/DADApy) (πŸ‘¨β€πŸ’» 19 Β· πŸ”€ 16 Β· πŸ“¦ 5 Β· πŸ“‹ 31 - 19% open Β· ⏱️ 02.07.2024): ``` git clone https://github.com/sissa-data-science/DADApy ``` - [PyPi](https://pypi.org/project/dadapy) (πŸ“₯ 220 / month): ``` pip install dadapy ```
ASAP (πŸ₯ˆ12 Β· ⭐ 140) - ASAP is a package that can quickly analyze and visualize datasets of crystal or molecular structures. MIT - [GitHub](https://github.com/BingqingCheng/ASAP) (πŸ‘¨β€πŸ’» 6 Β· πŸ”€ 28 Β· πŸ“¦ 6 Β· πŸ“‹ 25 - 24% open Β· ⏱️ 27.06.2024): ``` git clone https://github.com/BingqingCheng/ASAP ```
Show 5 hidden projects... - Sketchmap (πŸ₯ˆ8 Β· ⭐ 43 Β· πŸ’€) - Suite of programs to perform non-linear dimensionality reduction -- sketch-map in particular. GPL-3.0 C++ - Coarse-Graining-Auto-encoders (πŸ₯‰4 Β· ⭐ 21 Β· πŸ’€) - Unlicensed single-paper - paper-ml-robustness-material-property (πŸ₯‰4 Β· ⭐ 3 Β· πŸ’€) - BSD-3 datasets single-paper - KmdPlus (πŸ₯‰2 Β· ⭐ 3 Β· πŸ’€) - This module contains a class for treating kernel mean descriptor (KMD), and a function for generating descriptors with.. Unlicensed - Descriptor Embedding and Clustering for Atomisitic-environment Framework (DECAF) ( ⭐ 2) - Provides a workflow to obtain clustering of local environments in dataset of structures. Unlicensed


Visualization

Back to top

Projects that focus on visualization (viz.) for atomistic ML.

pymatviz (πŸ₯‡18 Β· ⭐ 140) - A toolkit for visualizations in materials informatics. MIT general-tool probabilistic - [GitHub](https://github.com/janosh/pymatviz) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 12 Β· πŸ“¦ 8 Β· πŸ“‹ 37 - 21% open Β· ⏱️ 01.07.2024): ``` git clone https://github.com/janosh/pymatviz ``` - [PyPi](https://pypi.org/project/pymatviz) (πŸ“₯ 960 / month): ``` pip install pymatviz ```
Chemiscope (πŸ₯‡18 Β· ⭐ 120) - An interactive structure/property explorer for materials and molecules. BSD-3 JavaScript - [GitHub](https://github.com/lab-cosmo/chemiscope) (πŸ‘¨β€πŸ’» 22 Β· πŸ”€ 29 Β· πŸ“₯ 220 Β· πŸ“¦ 6 Β· πŸ“‹ 120 - 28% open Β· ⏱️ 01.07.2024): ``` git clone https://github.com/lab-cosmo/chemiscope ``` - [npm](https://www.npmjs.com/package/chemiscope) (πŸ“₯ 7 / month): ``` npm install chemiscope ```
ZnDraw (πŸ₯‡18 Β· ⭐ 25) - A powerful tool for visualizing, modifying, and analysing atomistic systems. EPL-2.0 MD generative JavaScript - [GitHub](https://github.com/zincware/ZnDraw) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 1 Β· πŸ“¦ 3 Β· πŸ“‹ 240 - 22% open Β· ⏱️ 27.06.2024): ``` git clone https://github.com/zincware/ZnDraw ``` - [PyPi](https://pypi.org/project/zndraw) (πŸ“₯ 2K / month): ``` pip install zndraw ```


Wavefunction methods (ML-WFT)

Back to top

Projects and models that focus on quantities of wavefunction theory methods, such as Monte Carlo techniques like deep learning variational Monte Carlo (DL-VMC), quantum chemistry methods, etc.

DeepQMC (πŸ₯‡17 Β· ⭐ 330) - Deep learning quantum Monte Carlo for electrons in real space. MIT - [GitHub](https://github.com/deepqmc/deepqmc) (πŸ‘¨β€πŸ’» 13 Β· πŸ”€ 59 Β· πŸ“¦ 2 Β· πŸ“‹ 42 - 11% open Β· ⏱️ 23.02.2024): ``` git clone https://github.com/deepqmc/deepqmc ``` - [PyPi](https://pypi.org/project/deepqmc) (πŸ“₯ 51 / month): ``` pip install deepqmc ```
FermiNet (πŸ₯ˆ14 Β· ⭐ 650) - An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations. Apache-2 transformer - [GitHub](https://github.com/google-deepmind/ferminet) (πŸ‘¨β€πŸ’» 18 Β· πŸ”€ 110 Β· ⏱️ 04.06.2024): ``` git clone https://github.com/deepmind/ferminet ```
DeepErwin (πŸ₯‰11 Β· ⭐ 46) - DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions.. Custom - [GitHub](https://github.com/mdsunivie/deeperwin) (πŸ‘¨β€πŸ’» 7 Β· πŸ”€ 6 Β· πŸ“₯ 6 Β· ⏱️ 07.06.2024): ``` git clone https://github.com/mdsunivie/deeperwin ``` - [PyPi](https://pypi.org/project/deeperwin) (πŸ“₯ 30 / month): ``` pip install deeperwin ```
Show 1 hidden projects... - SchNOrb (πŸ₯‰5 Β· ⭐ 57 Β· πŸ’€) - Unifying machine learning and quantum chemistry with a deep neural network for molecular wavefunctions. MIT


Others

Back to top

pretrained-gnns (πŸ₯‡10 Β· ⭐ 940 Β· πŸ’€) - Strategies for Pre-training Graph Neural Networks. MIT pre-trained - [GitHub](https://github.com/snap-stanford/pretrain-gnns) (πŸ‘¨β€πŸ’» 2 Β· πŸ”€ 160 Β· πŸ“‹ 63 - 53% open Β· ⏱️ 29.07.2023): ``` git clone https://github.com/snap-stanford/pretrain-gnns ```
Show 1 hidden projects...

Contribution

Contributions are encouraged and always welcome! If you like to add or update projects, choose one of the following ways:

If you like to contribute to or share suggestions regarding the project metadata collection or markdown generation, please refer to the best-of-generator repository. If you like to create your own best-of list, we recommend to follow this guide.

For more information on how to add or update projects, please read the contribution guidelines. By participating in this project, you agree to abide by its Code of Conduct.

License

CC0