Made LSTM example nicer
Made LSTM example nicer

jackkamm/autograd,hips/autograd,t1m0thy/autograd,barak/autograd,HIPS/autograd,FighterLYL/autograd,kcarnold/autograd,melgor/autograd,hips/autograd,HIPS/autograd