div_valueΒΆ

Auto-generated from jax_code_defs

PyTorch

API: torch.nn.modules.adaptive.AdaptiveLogSoftmaxWithLoss.div_value
Strategy: Direct Mapping

JAX (Core)

API: optax.tree.div
Strategy: Direct Mapping

PaxML / Praxis

API: optax.tree.div
Strategy: Direct Mapping