Enable eqx.nn.State to track and return Intermediate values - #1165
Open
arenaudineau wants to merge 3 commits into
Open
Enable eqx.nn.State to track and return Intermediate values#1165arenaudineau wants to merge 3 commits into
arenaudineau wants to merge 3 commits into
