search

cgarciae / treex

A Pytree Module system for Deep Learning in JAX
https://cgarciae.github.io/treex/
MIT License
215 stars 17 forks source link

Removes double assignment in nn.Lambda #19

Closed ptigwe closed 2 years ago

cgarciae commented 2 years ago

Hey @ptigwe, thanks, nice catch!