Problems when replacing Flux by Lux in anodes example

Hello,
I’m trying to reproduce the anodes example on augmented neural ODEs from the thread Fitting neural ODE to periodic time series. However, when replacing Flux by the Lux library (with the corresponding changes) I get: FromFluxAdaptor requires Flux.jl to be loaded. I don’t have Flux added to Julia, and since I’m just starting to learn neural ODEs, I’m a bit lost here. Does anyone have any suggestions on how to modify the original code to use Lux instead of Flux?
Thanks!