polynomial-chaos
Global sensitivity analysis in JAX: Sobol, Morris, PCE, Shapley and ten more methods, vectorised and differentiable