Fix docs and simplify `dict_subset`
PEP8
Change attention transition interface
Change interaction between sequence generators and attention transitions
document `compute_states`
PEP8