warmup_cosine_decay_schedule¶ Auto-generated from jax_code_defs JAX (Core)Flax NNX JAX (Core)API: optax.warmup_cosine_decay_scheduleStrategy: Direct MappingOfficial Docs ↗Flax NNXAPI: optax.warmup_cosine_decay_scheduleStrategy: Direct MappingOfficial Docs ↗