WIP flax.linen flux.1 ported from nnx jflux#141
Closed
jcaraban wants to merge 3 commits intoAI-Hypercomputer:mainfrom
Closed
WIP flax.linen flux.1 ported from nnx jflux#141jcaraban wants to merge 3 commits intoAI-Hypercomputer:mainfrom
jcaraban wants to merge 3 commits intoAI-Hypercomputer:mainfrom