Improve docstrings

[jwdink]

One notable issue is usage of kwargs. For example, StateSpaceModel().fit() ultimately calls StateSpaceModel().forward(), and passing kwargs on to that method. This makes it difficult for someone who's working with fit() to know all their options (e.g. n_step).