Differentiable and accelerated spherical transforms with JAX
pytorch spherical-harmonics fourier-transform wigner-d-matrix jax differentiable-programming recursion-algorithm wigner-transform
-
Updated
Feb 3, 2025 - Python