Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX - View it on GitHub