N
NNX module
ConceptMentioned in 1 video
The core concept and fundamental building class in Flax NX, used to build layers, optimizers, and virtually everything else within the library.
The core concept and fundamental building class in Flax NX, used to build layers, optimizers, and virtually everything else within the library.
