rms_normΒΆ

Auto-generated from mlx_code_defs

PyTorch

API: torch.nn.functional.rms_norm
Strategy: Direct Mapping

Apple MLX

API: mlx.core.fast.rms_norm
Strategy: Direct Mapping