CrossEntropyLoss

Auto-generated from jax_code_defs

PyTorch

API: torch.nn.functional.cross_entropy
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: optax.softmax_cross_entropy_with_integer_labels
Strategy: Plugin (loss_reduction)