4 ms·
When I last investigated it a few months ago TensorFlow's TPU support was much more mature than PyTorch's. To get PyTorch to work on TPUs I had to download some
by bjourne 5y ago
When I last investigated it a few months ago TensorFlow's TPU support was much more mature than PyTorch's. To get PyTorch to work on TPUs I had to download some nightly builds and ask for help on Github since the tutorials weren't up to date. It also ran much slower than TensorFlow on Google Colab's TPUs. There are also some features in TensorFlow 2.0 I could not find any counterparts for in PyTorch, such as recurrent dropout for LSTM layers.