Skip to content

vesin v0.5.0

Latest

Choose a tag to compare

@github-actions github-actions released this 19 Feb 14:36

This release adds the ability to compute neighbor lists on GPU using CUDA!

The CUDA interface is accessible from C, C++, Fortran, Python and TorchScript, and automatically available for the pre-built wheels on PyPI.

vesin.NeighborList now accepts either numpy arrays, cupy arrays or torch tensor (with CPU or CUDA device) for the points, box, and periodic parameters.