In this PR we fix a bug in which wrong results are obtained when employing jax.jit using the static_argnums argument.
In particular, we use jax.jit now inside the function. To gain more certainty that LCM is producing the correct results, we've implemented an analytical test against we check the result of LCM. Furthermore, we fixed a missing assert statement in the regression test.
In this PR we fix a bug in which wrong results are obtained when employing
jax.jit
using thestatic_argnums
argument.In particular, we use
jax.jit
now inside the function. To gain more certainty that LCM is producing the correct results, we've implemented an analytical test against we check the result of LCM. Furthermore, we fixed a missingassert
statement in the regression test.