Composing Lux with Optim Algorithms

See Neural Ordinary Differential Equations · DiffEqFlux.jl. You can essentially use the Optimization.jl API to train Lux using Optim (and several other optimization packages out there)