Adafactor options: - no quantization - no sanitize_values via jnp.nan_to_num
Adafactor options:
- no quantization
- no sanitize_values via jnp.nan_to_num

PiperOrigin-RevId: 472779180

google/praxis,google/praxis