Differentiable and accelerated spherical transforms with JAX
pytorch spherical-harmonics fourier-transform wigner-d-matrix jax differentiable-programming recursion-algorithm wigner-transform
-
Updated
Dec 19, 2024 - Python