yuanz

icon company @catniplab @NIMH-MLT icon location machine learning core, nimh

Results 1 issues of yuanz

`WeightNorm` causes unexpected PyTrees inequality

8 comment

The following assertion fails. ``` import equinox as eqx import equinox.random as jrandom layer = eqx.nn.Linear(2, 2, key=jrandom.key(0)) m1 = eqx.nn.WeightNorm(layer) m2 = eqx.nn.WeightNorm(layer) assert eqx.tree_equal(m1, m2) # unequal ```...

feature