flax.linen

flax.linen#

Linen is the Flax Module system. Read more about our design goals in the Linen README.