I normally work with Tensorflow, but pytorch has become more popular in last few years