Improve @generated readability following SciML examples

[jeremiahpslewis]

e.g. https://github.com/SciML/DeepEquilibriumNetworks.jl/blob/2e9c5009c463f632c62c92100993c452ebf74c2b/src/utils.jl#L29