Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
VegasFlow: accelerating Monte Carlo simulation across multiple hardware platforms
An easy way to use multi-GPUs to calculate multi-dimensional integration