I have recently started using Flux to construct a physics informed neural network. To do this, I start with a model u = Chain(Dense(1, 10), Dense(10,15), Dense(15,1)) Then I construct a function to impose the differential condition y’(t) - y(t) = 0 function f(t) y = sum(u(t)) dy = gradient(() …

Using Flux for a neural net solution to differential equations

Elmo January 13, 2021, 4:16pm 5

Hi all, I also ran into this issue. I opened an issue on Flux and they solved it, see here for my discourse question and here for the answered flux issue.

Topic		Replies	Views
Flux differentiation error Machine Learning zygote	19	1685	November 19, 2020
"Can't differentiate foreigncall expression" error while using Flux.train! General Usage question , diffeq , flux , zygote	1	634	February 13, 2022
Gradient error in Flux model inputs Machine Learning question , flux , zygote	5	1324	January 13, 2021
Unable to solve a differential equation using neural network General Usage flux , neural-network	2	215	September 10, 2023
DifferentialEquations.jl + Flux.jl or Knet.jl Machine Learning diffeq	15	3597	October 18, 2018

Using Flux for a neural net solution to differential equations

Related topics