Using eqx.combine with static first causes problems with NonTrainable, as we get a None in params, which causes the structures to not match when combining. Briefly, this is because eqx.combine(None, [1,1]) works, and eqx.combine([1,1], None) is not allowed as the "structures do not match".
Using
eqx.combinewith static first causes problems withNonTrainable, as we get aNoneinparams, which causes the structures to not match when combining. Briefly, this is becauseeqx.combine(None, [1,1])works, andeqx.combine([1,1], None)is not allowed as the "structures do not match".