Gradient clipping model extension