Germán Abrevaya

SDEs gradients issues with EnsembleProblems

4

Strangely, the following codes work well if DiffEqFlux is not loaded but they break if it is. MWE ```julia using Zygote using StochasticDiffEq using SciMLSensitivity using DiffEqFlux # if used,...

Type instability on gradients possibly involving indexing of vectors

1

Despite `h` being type-stable, its gradient is not. ```julia using Zygote using Test f(x) = [x] g(x) = f(x)[1] h(x) = sum(g(x)) v = rand(Float32, 3, 2) @inferred h(v) #...

Memory leak with pyplot

1

When I create matplotlib.pyplot plots, the memory keeps increasing and is not released by the garbage collector. The following is the case I'm interested in (with subplots) and bellow I...

Hi! I got a segmentation fault while attempting to differentiate a function involving an EnsembleProblem with StaticArrays. I managed to isolate the error in this MWE: ```julia using Zygote using...

Replacing the EnsembleProblem with an equivalent simple for loop breaks the training

2

If we change the differential equations solving inside the `diffeq_layer` from the current `EnsembleProblem` to a simple for loop, which forward pass gives the exact same result, the whole training...

Logging plots causes memory leak

1

Hi, I've just realized that the cause of the memory leak that I was facing was due to logging (maybe in the wrong way) with Wandb.jl I was using it...