Docs improvememnts, more flexible Keras dependency, drop typechecks dependency, remove some unused functions