Skip to main content

Browse the archive

https://github.com/JuliaDiffEq/DiffEqFlux.jl
13 April 2020, 12:04:22 UTC
Revision 69b327661be2b80299b80bc71f965880546efb90 authored by Christopher Rackauckas on 09 April 2019, 10:58:13 UTC, committed by GitHub on 09 April 2019, 10:58:13 UTC 
Merge pull request #37 from JuliaDiffEq/zero_gradients
 
Fix zero'd gradients from save_start
2 parent s 6013056 + c2380fc
Raw File
Tip revision: 69b327661be2b80299b80bc71f965880546efb90 authored by Christopher Rackauckas on 09 April 2019, 10:58:13 UTC
Merge pull request #37 from JuliaDiffEq/zero_gradients
Tip revision: 69b3276
DiffEqFlux.jl 
module DiffEqFlux

using DiffEqBase, Flux, DiffResults, DiffEqSensitivity, ForwardDiff

include("Flux/layers.jl")
include("Flux/neural_de.jl")
include("Flux/utils.jl")

export diffeq_fd, diffeq_rd, diffeq_adjoint
export neural_ode, neural_ode_rd
export neural_dmsde
end
back to top