SparseCategoricalCrossentropy

Computes the sparse categorical crossentropy loss.

Abstract Signature:

SparseCategoricalCrossentropy(y_true: Tensor, y_pred: Tensor, from_logits: bool = False, ignore_class: int)

PyTorchJAX (Core)KerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.nn.functional.cross_entropy

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: optax.softmax_cross_entropy_with_integer_labels

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.losses.sparse_categorical_crossentropy

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.keras.losses.sparse_categorical_crossentropy

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.losses.cross_entropy

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: optax.softmax_cross_entropy_with_integer_labels

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: optax.softmax_cross_entropy_with_integer_labels

Strategy: Direct Mapping

Official Docs ↗