Zero grad at start of loop instead of end in engines to avoid issues

Zero grad at start of loop instead of end in engines to avoid issues #58