Make "train" dependencies optional

[vmarkovtsev]

Some of our Python dependencies are only used for pretraining some models, e.g. typos. We need to add a "train" option to setup.py so that they are not installed by default.