OpenSourceEconomics / lcm

Solution and simulation of life cycle models in Python with GPU acceleration.
Apache License 2.0
16 stars 1 forks source link

Fix jitting bug #51

Closed timmens closed 11 months ago

timmens commented 11 months ago

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.