Main changes:

• Full PyTorch support
• Automated hyper parameter tuning module
• Added validation set for hyper parameter tuning
• Several important bug fixes
• New baselines.