scale_grad_by_freq ================== Auto-generated from paxml_code_defs .. raw:: html

PyTorch

API: torch.nn.modules.sparse.EmbeddingBag.scale_grad_by_freq
Strategy: Direct Mapping

PaxML / Praxis

API: optax.tree.scale
Strategy: Direct Mapping