Closed avik-pal closed 2 weeks ago
Currently it errors if the result has a NamedTuple. All Lux models return a state (NamedTuple), so it would be nice to add a case for that.
Maybe https://github.com/EnzymeAD/Reactant.jl/pull/6/files already does this?
Currently it errors if the result has a NamedTuple. All Lux models return a state (NamedTuple), so it would be nice to add a case for that.