aksuhton

Stacktrace[38:end] ``` [38] _pullback(ctx::Zygote.Context{false}, f::Zygote.Pullback{Tuple{typeof(jacobian), StatefulLuxLayer{true, Dense{true, typeof(gelu), typeof(glorot_uniform), typeof(zeros32)}, @NamedTuple{weight::Matrix{Float32}, bias::Matrix{Float32}}, @NamedTuple{}}, Matrix{Float32}}, Tuple{Zygote.ZBack{LuxZygoteExt.var"#13#17"{StatefulLuxLayer{true, Dense{true, typeof(gelu), typeof(glorot_uniform), typeof(zeros32)}, @NamedTuple{weight::Matrix{Float32}, bias::Matrix{Float32}}, @NamedTuple{}}, Matrix{Float32}, @NamedTuple{weight::Matrix{Float32}, bias::Matrix{Float32}}, Tuple{Matrix{Float32}}}}, Zygote.var"#2180#back#303"{Zygote.var"#back#302"{:ps, Zygote.Context{false}, StatefulLuxLayer{true, Dense{true,...

Thank you @avik-pal, sorry for the 😓.

Hey @tansongchen, thanks for joining us! Absolutely. I'm trying to learn a mapping between two time-sequences $\varepsilon_t \rightarrow \sigma_t$. Now, thermodynamic arguments suggest that $\sigma_t = \frac{\partial F_t}{\partial \varepsilon_t}$ for...

Did this ever reach a conclusion? I'm in need of the ability to take the jacobian with respect to the inputs of a (Lux) model output and then optimize that...

OP's code works now on a CPU but fails when I try to do this on the GPU. I'd like this functionality but as a relatively new programmer I'm not...

Commenting merely to give thanks for the container instructions.

Returning here to say that I am finding success with out of the box AMDGPU.jl Thank you everyone who made this possible