torchquad
torchquad copied to clipboard
Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
# Issue ## Problem Description Current usage may overwrite loguru logger in other applications ## Expected Behavior Not interfering with user's logging ## What Needs to be Done Correct implementation...
# Issue ## Problem Description Currently tests throw a lot of warnings on `develop` branch ``` =============================== warnings summary =============================== torchquad/tests/boole_test.py: 4 warnings torchquad/tests/gradient_test.py: 3 warnings torchquad/tests/integrator_types_test.py: 4 warnings torchquad/tests/monte_carlo_test.py:...
This PR changes the import statement for `jax.config`, which resolves the error ``` File "/usr/local/lib/python3.10/dist-packages/torchquad/utils/set_precision.py", line 58, in set_precision from jax.config import config ImportError: cannot import name 'config' from 'jax.config'...
# Issue ## Problem Description When using keyword arguments in `Simpson().calculate_result()` function, fails with `IndexError: tuple index out of range`. The issue seems to come from `expand_func_values_and_squeeze_integral` only checking `args`...