Synchronize training dependency versions between Docker image and wheel, update docs, refactor build scripts.