NativeBatchNorm

Native batch normalization (often internal, maps to standard BN).

Abstract Signature:

NativeBatchNorm(input: Tensor, weight: Tensor, bias: Tensor, running_mean: Tensor, running_var: Tensor, training: bool, momentum: float, eps: float)

PyTorchJAX (Core)Keras

PyTorch

API: torch.native_batch_norm

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.nn.standardize

Strategy: Plugin (batch_norm_unwrap)

Keras

API: keras.ops.batch_norm

Strategy: Direct Mapping

Official Docs ↗