GitHub repository: lab-cosmo/torch_spex
Primary language: Python
Description: Spherical expansions in PyTorch