Cleaned up neural net example a little
Cleaned up neural net example a little

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