InstanceNorm

Applies Instance Normalization.

Abstract Signature:

InstanceNorm(input: Tensor, weight: Tensor, bias: Tensor, running_mean: Tensor, running_var: Tensor, use_input_stats: bool = True, momentum: float = 0.1, eps: float = 1e-05)

PyTorchJAX (Core)KerasApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.instance_norm

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

Keras

API: keras.ops.moments

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.InstanceNorm

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: flax.nnx.InstanceNorm

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: praxis.layers.InstanceNorm

Strategy: Direct Mapping

Official Docs ↗