cuEquivariance is an NVIDIA Python library designed to facilitate the construction of high-performance geometric neural networks using segmented polynomials and triangular operations. cuEquivariance provides a comprehensive API for describing segmented polynomials made out of segmented tensor…