Rémi Louf
Rémi Louf
Adding to this, you can set any reasonable IDE up so when you run tests it will open a debugger console whenever it hits a breakpoint or fails. If you...
PRs that touch on core mechanisms in Aesara, or simply that implement big changes, can easily get frustrating. Breaking the problem down like you did is a great reaction to...
Glad to hear this is back on track!
@junpenglao ping
Is there anything outstanding here @LegrandNico, besides the conflicts?
> I can work on it a bit later today. I had difficulties creating a new action and I was not sure if this should be a part of the...
We can actually work around a lot of the dynamic indexing issues with `jax.lax.dynamic_slice` so I'm currently fixing this throughout the dispatcher. The basic test with the fixed number of...
After a lot of messing around I decided to go for a full rewrite and follow the Numba implementation. I have a minimal version that passes the first 3 `xit_xot_types`...
This is turning into a much bigger PR than expected as I am also trying to fix any issue that prevents me from running the `scan` tests. Left to be...
The following test with a `RandomStream` fails: ```python def test_nit_sot_shared(): res, updates = scan( fn=lambda: RandomStream(seed=1930, rng_ctor=np.random.RandomState).normal( 0, 1, name="a" ), n_steps=3, ) jax_fn = function((), res, updates=updates, mode="JAX") jax_res...