@aaronsteven managing state collection for big mod…

@aaronsteven managing state collection for big models like LSTM actually pretty tricky, & can be (partially) abstracted; see prettytensor