scale_grad_by_freq

Auto-generated from paxml_code_defs

PyTorch

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

PaxML / Praxis

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