torch_featurize API Reference
================================

The ``aenet.torch_featurize`` module provides PyTorch-based implementations
of atomic environment descriptors and neighbor lists.

Module Overview
---------------

.. currentmodule:: aenet.torch_featurize

Main Classes
~~~~~~~~~~~~

.. autosummary::
   :toctree: generated/

   ChebyshevDescriptor
   BatchedFeaturizer
   TorchNeighborList

Basis Functions
~~~~~~~~~~~~~~~

.. autosummary::
   :toctree: generated/

   ChebyshevPolynomials
   RadialBasis
   AngularBasis