jonasguan
jonasguan
Thanks for the suggestion @jpgard . Would you be interested in submitting a PR for this?
Thanks for the suggestion @ZuowenWang0000! If you can submit a PR with your proposed changes, we'd be glad to review and merge it.
Oops, did not mean to close this yet.
Hey @andantillon , sorry for the late reply, we had been a bit inactive for a while, but we're launching version 4 of the library soon and will be more...
You have to mark `training` as a static argument when you jit your functions, so the compiler knows that you're ok with recompiling the function if its value changes. See:...