esa / torchquad

Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX

GNU General Public License v3.0

189 stars 40 forks source link

Closed aurelio-amerio closed 3 weeks ago

aurelio-amerio commented 3 weeks ago

Description

I believe from jax.config import config is a typo and it should be changed to from jax import config.

Summary of changes

aurelio-amerio commented 3 weeks ago

I hadn't realized there was a PR going on