The one advantage of pytorch is that you can make your graphs dynamic.