max_norm

Auto-generated from jax_code_defs

PyTorch

API: torch.nn.modules.sparse.EmbeddingBag.max_norm
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: optax.tree.max
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: optax.tree.max
Strategy: Direct Mapping
Official Docs ↗