I understand how the heavier weights can cause you to lock out that back knee. I notice myself wanting to do the same thing. I would suggest doing jerk balances and split push presses behind the neck or in front. Focus on keeping your back knee soft and your torso upright. We do these occasionally to warm up before we jerk.
You can see an example in this video of Jessica doing a combo to warm up: