tidy up
test parallel SGD
profile parallel SGD