Describe the bug
Running the paper example, on the latest version of jax 0.4.9, generates some high-frequency oscillations in the field that were not present before.
To Reproduce
Steps to reproduce the behavior:
run the example_1_paper notebook
Plot the solution field for the helmholtz equation
Expected behavior
The new oscillations simply should not be there.
Describe the bug Running the paper example, on the latest version of
jax 0.4.9
, generates some high-frequency oscillations in the field that were not present before.To Reproduce Steps to reproduce the behavior:
example_1_paper
notebookExpected behavior The new oscillations simply should not be there.