PyTorch
API: torch.nn.functional.mse_loss
Strategy: Direct Mapping
JAX (Core)
API: optax.squared_error
Strategy: Direct Mapping
Keras
API: keras.losses.mean_squared_error
Strategy: Direct Mapping
TensorFlow
API: tf.keras.losses.mean_squared_error
Strategy: Direct Mapping
Apple MLX
API: mlx.nn.losses.mse_loss
Strategy: Direct Mapping
Flax NNX
API: optax.squared_error
Strategy: Direct Mapping
PaxML / Praxis
API: optax.squared_error
Strategy: Direct Mapping