Skip to main content

Scalable EquiVariance Enabled Neural Network

Project description

Alt text

SevenNet

SevenNet (Scalable EquiVariance-Enabled Neural Network) is a graph neural network (GNN)-based interatomic potential package that supports parallel molecular dynamics simulations using LAMMPS. Its core model is based on NequIP.

Full documentation, including installation, usage, and pretrained models, is available at documentation.

Features

  • Pretrained GNN interatomic potential and fine-tuning interface
  • Atomic Simulation Environment (ASE) calculator (python)
  • GPU-parallelized molecular dynamics with LAMMPS
  • CUDA-accelerated D3 (van der Waals) dispersion
  • Multi-fidelity training for combining multiple databases with different calculation settings
  • Fine-tuning with forgetting prevention (experience replay + Elastic Weight Consolidation) for continual learning
  • Tensor product accelerators

Pretrained Models

SevenNet provides pretrained models (universal potentials). Please refer to the documentation for available checkpoints, and usage examples: Pretrained models

Installation and user guides

Installation (including LAMMPS and D3) and user guides can be found in our documentation.

The old README (prior to v0.12.0) can be found here.

Citation

If you use this code, please cite:

@article{park_scalable_2024,
	title = {Scalable Parallel Algorithm for Graph Neural Network Interatomic Potentials in Molecular Dynamics Simulations},
	volume = {20},
	doi = {10.1021/acs.jctc.4c00190},
	number = {11},
	journal = {J. Chem. Theory Comput.},
	author = {Park, Yutack and Kim, Jaesun and Hwang, Seungwoo and Han, Seungwu},
	year = {2024},
	pages = {4857--4868},
}

If you utilize the multi-fidelity feature of this code or the pretrained model SevenNet-MF-ompa, please cite the following paper:

@article{kim_sevennet_mf_2024,
	title = {Data-Efficient Multifidelity Training for High-Fidelity Machine Learning Interatomic Potentials},
	volume = {147},
	doi = {10.1021/jacs.4c14455},
	number = {1},
	journal = {J. Am. Chem. Soc.},
	author = {Kim, Jaesun and Kim, Jisu and Kim, Jaehoon and Lee, Jiho and Park, Yutack and Kang, Youngho and Han, Seungwu},
	year = {2024},
	pages = {1042--1054},
}

If you utilize the pretrained model SevenNet-Omni or multi-task training strategies including task-specific regularization and domain-bridging dataset, please cite the following paper:

@article{kim_optimizing_2025,
	title = {Optimizing Cross-Domain Transfer for Universal Machine Learning Interatomic Potentials},
	volume = {17},
	doi = {10.1038/s41467-026-70195-8},
	number = {3432},
	journal = {Nat. Commun.},
	author = {Kim, Jaesun and You, Jinmu and Park, Yutack and Lim, Yunsung and Kang, Yujin and Kim, Jisu and Jeon, Haekwan and Ju, Suyeon and Hong, Deokgi and Lee, Seung Yul and Choi, Saerom and Kim, Yongdeok and Lee, Jae W and Han, Seungwu},
	year = {2026},
}

If you utilize the reEWC forgetting-aware fine-tuning strategy for continual learning of pretrained universal machine-learning interatomic potentials, please cite the following paper:

@article{kim_efficient_2026,
	title = {An Efficient Forgetting-Aware Fine-Tuning Framework for Pretrained Universal Machine-Learning Interatomic Potentials},
	volume = {12},
	doi = {10.1038/s41524-025-01895-w},
	number = {26},
	journal = {npj Comput. Mater.},
	author = {Kim, Jisu and Lee, Jiho and Oh, Sangmin and Park, Yutack and Hwang, Seungwoo and Han, Seungwu and Kang, Sungwoo and Kang, Youngho},
	year = {2026},
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sevenn-0.13.0.tar.gz (43.1 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

sevenn-0.13.0-py3-none-any.whl (43.2 MB view details)

Uploaded Python 3

File details

Details for the file sevenn-0.13.0.tar.gz.

File metadata

  • Download URL: sevenn-0.13.0.tar.gz
  • Upload date:
  • Size: 43.1 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.10

File hashes

Hashes for sevenn-0.13.0.tar.gz
Algorithm Hash digest
SHA256 3f8c217182dc3d1bc3578693c388f386afd15e1c9affd4fc1fda6732f384d6c5
MD5 ebbbbf54ad814ce9ef4025f6e0bf1cc6
BLAKE2b-256 baf8884e3b488beb80439a14edf33a5d6b56a99c0f95df5e57eb46e44daf11e0

See more details on using hashes here.

File details

Details for the file sevenn-0.13.0-py3-none-any.whl.

File metadata

  • Download URL: sevenn-0.13.0-py3-none-any.whl
  • Upload date:
  • Size: 43.2 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.10

File hashes

Hashes for sevenn-0.13.0-py3-none-any.whl
Algorithm Hash digest
SHA256 9378f7df2b75fe83274483a5f804790ae1023dd22ddeb6a48c1a03de97e22620
MD5 3d6fb9e10bdabed59f0733b32e6c4e55
BLAKE2b-256 c494290a417e4b5658f1a7ee8fba86ba9e53d5341cce5feaa3024b454facfd5e

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page