Gradient Based Optimization Code Python Game Snake Google