Jad Mounayer

fixing bugs 323 and 324

2

I fixed both bugs 323 and 324 and tested linear, rectilinear and cubic interpolations with both empty and random matrices. Tests are passing but I am getting 4 warnings (which...

replace_nans_at_start overwrite existing values

1

Hello, I am not sure if the behavior of **replace_nans_at_start** in **backward_hermite_coefficients** is expected when we have initial values that aren't all NaNs. The documentation proposes that "replace_nans_at_start" will change...

Treating Channels that only have missing values in interpolations

1

While using **"diffrax.backward_hermite_coefficients"**, I noticed that the interpolation technique doesn't take into consideration the case when all the entries are missing (i.e. NaN), even if **"replace_nans_at_start"** is specified. I beleive...

"masked" leads to unexpected behavior with callable Pytrees

6

Good morning! The "masked" function finds the "mask_tree" as follows: mask_tree = mask(params) if callable(mask) else mask (1) Which is used twice (in "init_fn" and "update_fn"). However, in some cases,...