Rapid research framework for PyTorch. The researcher's version of Keras - View it on GitHub