Assertion: flax | /r/CapableOf | manage batchnorm statistics
Facets:
value: with scopes
type: OTHER
Source sentences:
text: Flax uses with scopes to manage state (like batchnorm statistics) and JAX’s functional RNGs for stochastic layers (like dropout).