PyTorch
API: —
Strategy: Macro 'torch.log(torch.cosh({y_pred} - {y_true}))'
JAX (Core)
API: optax.log_cosh
Strategy: Direct Mapping
Keras
API: keras.losses.log_cosh
Strategy: Direct Mapping
TensorFlow
API: tf.keras.losses.log_cosh
Strategy: Direct Mapping
Apple MLX
API: mlx.nn.losses.log_cosh_loss
Strategy: Direct Mapping
Flax NNX
API: optax.log_cosh
Strategy: Direct Mapping
PaxML / Praxis
API: optax.log_cosh
Strategy: Direct Mapping