Module utils 

Modules

log_sigmoid

loss

rms_norm

rms_norm_gated

sanity

scheduler

Learning rate schedulers for controlling the optimization process.

silu

softplus

Functions

div_eps

div_eps_f32

stable_max