TensorFlow version of the model was contributed by ariG23498,
gante, and sayakpaul (equal contribution).