Tldr; https://github.com/sunlabuiuc/PyHealth/blob/master/pyhealth/models/rnn.py https://github.com/sunlabuiuc/PyHealth/blob/c5c01d9a9347d1779fa1be98cacd0098e1ee047d/pyhealth/models/rnn.py#L239