NNX module

Concept

The core concept and fundamental building class in Flax NX, used to build layers, optimizers, and virtually everything else within the library.

Mentioned in 1 video

Videos Mentioning NNX module

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

DeepLearningAI

The core concept and fundamental building class in Flax NX, used to build layers, optimizers, and virtually everything else within the library.