add_decayed_weights =================== Auto-generated from jax_code_defs .. raw:: html
optax.add_decayed_weights