BatchStat

Wrapper for Batch Statistics state.

PyTorch

API: —
Strategy: Custom / Partial

Keras

API: keras.Variable
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.BatchStat
Strategy: Direct Mapping
Official Docs ↗