Public neural-network API surface.
This import point exposes blocks, heads, ResNet/Vision layers, and Transformer components through
the user-facing NN.API.nn namespace.
Public neural-network API surface.
This import point exposes blocks, heads, ResNet/Vision layers, and Transformer components through
the user-facing NN.API.nn namespace.