This commit adds a new test for testing session state saving and
loading using in-memory models. It currently tests a normal transformer
model, a recurrent model, and a hybrid model.
The motivation for this is to be able to run these test in CI without
having to depend on real models or depend on manual testing.