splitting the linear maps is faster and uses less memory in my case
splitting the linear maps is faster and uses less memory in my case

kbullaughey/lstm-play,kbullaughey/lstm-play,kbullaughey/lstm-play