I think it would be cool to also have the core gradient operation based problems like:

• Gradient Clipping
• Gradient Accumulation
• Gradient Checkpointing

Can work on these too @moe18