GitHub repository: lab-cosmo/torch-spex
Primary language: Python
Description: spherical expansions in torch