Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/luthaf/vesin

Compute neighbor lists for atomistic systems
https://github.com/luthaf/vesin

Last synced: 4 days ago
JSON representation

Compute neighbor lists for atomistic systems

Awesome Lists containing this project

README

        

# Vesin: fast neighbor lists for atomistic systems

[![Documentation](https://img.shields.io/badge/docs-latest-brightgreen.svg)](http://luthaf.fr/vesin/)
![Tests](https://img.shields.io/github/check-runs/Luthaf/vesin/main?logo=github&label=tests)

| English ๐Ÿ‡บ๐Ÿ‡ธโ /โ ๐Ÿ‡ฌ๐Ÿ‡ง | Occitan | French ๐Ÿ‡ซ๐Ÿ‡ท | Arpitan | Galloโ€‘Italic | Catalan | Spanish ๐Ÿ‡ช๐Ÿ‡ธ | Italian ๐Ÿ‡ฎ๐Ÿ‡น |
|------------------|----------|-----------|----------|--------------|---------|------------|------------|
| neighbo(u)r | vesin | voisin | vesin | visin | veรญ | vecino | vicino |

Vesin is a fast and easy to use library computing neighbor lists for atomistic
system. We provide an interface for the following programing languages:

- C (also compatible with C++). The project can be installed and used as a
library with your own build system, or included as a single file and built
directly by your own build system;
- Python;
- TorchScript, with both a C++ and Python interface;

### Installation

To use the code from Python, you can install it with `pip`:

```
pip install vesin
```

See the [documentation](https://luthaf.fr/vesin/latest/index.html#installation)
for more information on how to install the code to use it from C or C++.

### Usage instruction

You can either use the `NeighborList` calculator class:

```py
import numpy as np
from vesin import NeighborList

# positions can be anything compatible with numpy's ndarray
positions = [
(0, 0, 0),
(0, 1.3, 1.3),
]
box = 3.2 * np.eye(3)

calculator = NeighborList(cutoff=4.2, full_list=True)
i, j, S, d = calculator.compute(
points=points,
box=box,
periodic=True,
quantities="ijSd"
)
```

We also provide a function with drop-in compatibility to ASE's neighbor list:

```py
import ase
from vesin import ase_neighbor_list

atoms = ase.Atoms(...)

i, j, S, d = ase_neighbor_list("ijSd", atoms, cutoff=4.2)
```

See the [documentation](https://luthaf.fr/vesin/latest/c-api.html) for more
information on how to use the code from C or C++.

### Benchmarks

You can find below benchmark result computing neighbor lists for increasingly
large diamond supercells, using an AMD 3955WX CPU and an NVIDIA 4070 Ti SUPER
GPU. You can run this benchmark on your system with the script at
`benchmarks/benchmark.py`. Missing points indicate that a specific code could
not run the calculation (for example, NNPOps requires the cell to be twice the
cutoff in size, and can't run with large cutoffs and small cells).

![Benchmarks](./docs/src/benchmark.png)

## License

Vesin is is distributed under the [3 clauses BSD license](LICENSE). By
contributing to this code, you agree to distribute your contributions under the
same license.