Add Gradient Crop

Only tested on CUDA, MUSA compatibility unknown. If the loss still takes off, try changing the second parameter of ```clip_grad_value_(model.parameters(), 1)``` to something smaller (e.g. 1 to 0.8)