[20:01:21] - INFO - absl - A polynomial schedule was set with a non-positive `transition_steps` value; this results in a constant schedule with value `init_value`.
Traceback (most recent call last):
  File "./run_mlm_flax.py", line 688, in <module>
    optax.clip_grad_by_global_norm(1.),
AttributeError: module 'optax' has no attribute 'clip_grad_by_global_norm'