# API Reference

This is the API reference for NVIDIA ALCHEMI Toolkit-Ops. Please refer to the full user
guide and examples for further details on how to use this package as API documentation
alone may not provide sufficient information.

## PyTorch Reference

```{toctree}
:maxdepth: 2

torch/neighbors.rst
torch/segment_ops.rst
torch/dispersion.rst
torch/electrostatics.rst
torch/dynamics.rst
torch/autograd.rst
```

## JAX Reference

```{toctree}
:maxdepth: 2

jax/neighbors.rst
jax/segment_ops.rst
jax/dispersion.rst
jax/electrostatics.rst
```

## Warp Reference

```{toctree}
:maxdepth: 2

warp/neighbors.rst
warp/segment_ops.rst
warp/dispersion.rst
warp/electrostatics.rst
warp/dynamics.rst
utilities.rst
```