Refactor experiment/train.py to main.py

We start all the experiments from experiment/train.py. It is our main script. Rename it to main.py and put it at the repository root.