Thibault Monsel issues

Results 5 issues of


                                            Thibault Monsel

Possibility of calling parts of ```apply``` functions of a transform ```hk``` module ?

Hello, I would like to use simultaneously my ```apply``` function of my ```Model``` haiku module and also the ```apply``` of the submodule ```self.mlp1``` with the same variable. I was wondering...

First DDE version

New files : - `discontinuity.py` that does the root finding during integration steps. Modified files: - `integrate.py` changed a bit of the code but essentially looks the same but with...

Vmapping over callables / pytest cache_clear() error

Hi @patrick-kidger, # `vmap` Callables I'd like to use a `vmap`'d array of `Callable` functions inside a `dummy_diffeq` that would also be `vmap`'d A MWE would be something of the...

question

`Dopri5` solver doesn't work for DDEs

Hello I was playing around with the DDE solver on some equations and consistently Dopri5 doesn't integrate properly. MWE : ```jl using DifferentialEquations using DelimitedFiles using PyPlot using Metrics #...

`EQX_ON_ERROR=breakpoint` and `diffeqsolve` causes tracer leakage ?

Hi Patrick, i'm getting weird behavior with HEAD and its use with `EQX_ON_ERROR=breakpoint`. Here is a MWE : ```python import jax import jax.numpy as jnp import diffrax with jax.checking_leaks(): ts...

question