SciML/DiffEqFlux.jl
View on GitHubImplement Local Regularization of Neural DEs in DiffEqFlux
Open
#817 opened on Mar 6, 2023
good first issue
Repository metrics
- Stars
- (919 stars)
- PR merge metrics
- (Avg merge 1d 14h) (11 merged PRs in 30d)
Description
Here's the current implementation -- https://github.com/avik-pal/LocalRegNeuralDE.jl, which is not super user-friendly (and is unlikely that I will register it). All we need to do is:
- Implement perform step for an ODE solver -- Tsit5 works well
- Implement perform step for an SDE solver -- EulerHeun maybe?
- Add dispatches for the specific neural de layers similar to https://github.com/avik-pal/LocalRegNeuralDE.jl/blob/main/src/layers/neural_ode.jl