Thibault Monsel

Results 23 comments of Thibault Monsel

Hello there, As mentionned by @patrick-kidger, most of the code itself is there (I would say 90%) and functional but some `jax` related bugs are still there (e.g. tracer leakage)...

That's great to here ! Integrating DDEs itself should be more than robust (so `dde.ipynb` should work fine). However, fitting a DDE with a neural net will shootout some `Exception:...

Thanks for the clear explanation ! I'll give `JAX_DISABLE_JIT=1` a try. This bug is definitely misleading.