monte-carlo-integration
Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
An easy way to use multi-GPUs to calculate multi-dimensional integration
Monte Carlo integration with Tensorflow
