Direct model lstm

**Requirement Text**
Update lstm to support direct model (i.e., time_of_event)

**Background Information**
Currently lstm is just state transition

**DoD**
 - [ ] Implement feature
 - [ ] Add to example 
 - [ ] Add tests 
 - [ ] update documentation
 - [ ] Add to change notes for next release