Use autoray's numpy mimic instead of torch for tensor operations

FHof / torchquad

Multidimensional numerical integration on the GPU using PyTorch

GNU General Public License v3.0

0 stars 0 forks source link

Closed FHof closed 2 years ago

FHof commented 3 years ago

Numpy and JAX as numerical backends are now supported in Trapezoid, Simpson, Boole and MonteCarlo.

FHof commented 2 years ago

I've added 4 new commits:

backend parameter for the integrate methods
random number generator class for uniform random number generation in MonteCarlo
one-line change to support tensorflow in Newton Cotes rules. This only works if the user enables the numpy behaviour:
```
from tensorflow.python.ops.numpy_ops import np_config
np_config.enable_numpy_behavior()
```
two-line change to remove unused IntegrationGrid h initialisation