Error in trying to use Optimization.jl for LSTM training based on Lux.jl

Error in trying to use Optimization.jl for LSTM training based on Lux.jl #5440