Metadata-Version: 2.4
Name: torch_admp
Version: 1.1.5
Summary: ADMP in PyTorch backend
License: SPDX-License-Identifier: LGPL-3.0-or-later
        
Project-URL: Homepage, https://github.com/ChiahsinChu/torch-admp
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch<2.10,>=2.1.0
Requires-Dist: torchopt==0.7.3
Requires-Dist: numpy>=1.21
Requires-Dist: scipy>=1.6.0
Requires-Dist: ase
Requires-Dist: ml_dtypes
Provides-Extra: vesin
Requires-Dist: vesin[torch]; extra == "vesin"
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: pytest-html; extra == "test"
Requires-Dist: mypy; extra == "test"
Requires-Dist: openmm>=7.7.0; extra == "test"
Requires-Dist: jax<=0.4.20,>=0.4.1; extra == "test"
Requires-Dist: jaxlib<=0.4.20,>=0.4.1; extra == "test"
Requires-Dist: numpy<2.0,>=1.21; extra == "test"
Requires-Dist: freud-analysis==3.4.0; extra == "test"
Provides-Extra: docs
Requires-Dist: mkdocs>=1.5.0; extra == "docs"
Requires-Dist: mkdocs-material>=9.0.0; extra == "docs"
Requires-Dist: mkdocstrings[python]>=0.20.0; extra == "docs"
Requires-Dist: mkdocs-glightbox>=0.3.0; extra == "docs"
Requires-Dist: pymdown-extensions>=10.0.0; extra == "docs"
Requires-Dist: mkdocs-git-revision-date-localized-plugin>=1.2.0; extra == "docs"
Requires-Dist: mkdocs-awesome-pages-plugin>=2.8.0; extra == "docs"
Requires-Dist: mkdocs-gen-files>=0.5.0; extra == "docs"
Requires-Dist: griffe>=0.48.0; extra == "docs"
Dynamic: license-file

# ADMP (Automatic Differentiable Multipolar Polarizable) in PyTorch backend

[![Zenodo doi badge](https://img.shields.io/badge/DOI-10.5281%2Fzenodo.18372172-blue.svg)](https://doi.org/10.5281/zenodo.18372172)

[![codecov](https://codecov.io/gh/ChiahsinChu/torch-admp/graph/badge.svg?token=aBlN5QoejV)](https://codecov.io/gh/ChiahsinChu/torch-admp)
[![docscov](./docs/badges/docstring-coverage.svg)](https://pypi.org/project/docstr-coverage/)
[![PyPI version](https://img.shields.io/pypi/v/torch-admp)](https://pypi.org/project/torch-admp/)

> torch version of ADMP is initialized by [Zheng Cheng](https://github.com/zhengcheng233/dmff_torch) (AISI).

This package implements the PME method (for monopoles) and the QEq method in [DMFF](https://github.com/deepmodeling/DMFF) with PyTorch, allowing not only GPU-accelerated calculation of PME/QEq methods but also further customization and extension of other PyTorch-based models.

## Installation

This package can be installed by:

```bash
pip install torch-admp
```

For the unit tests, you can install the package from source with the following command:

```bash
git clone https://github.com/ChiahsinChu/torch-admp
pip install torch-admp[test,vesin]
pip install "DMFF @ git+https://github.com/ChiahsinChu/DMFF.git@devel"
```

## [Documentation](https://chiahsinchu.github.io/torch-admp/)

## License

This project is licensed under the LGPL-3.0-or-later license. See the [LICENSE](LICENSE) file for details.

## Citation

If you use torch-admp in your research, please refer to the full citation details in the [CITATION.cff](CITATION.cff) file.

## Contributing

Contributions are welcome! Please feel free to submit a Pull Request. For major changes, please open an issue first to discuss what you would like to change.

## Support

For questions and support, please open an issue on the [GitHub repository](https://github.com/ChiahsinChu/torch-admp).
