PyTorch and Tensorflow functional model definitions - View it on GitHub