The Living Thing / Notebooks :



Usefulness: 🔧
Novelty: 💡
Uncertainty: 🤪 🤪 🤪
Incompleteness: 🚧 🚧 🚧

Successor to Lua’s torch. Evil twin to googles’s Tensorflow. Probably ascendant over Tensorflow for researchers if not for industrial uses.

They claim to be aiming for fancy features such as reversible learning and what-have-you.

PyTorch has a unique way of building neural networks: using and replaying a tape recorder.

Most frameworks such as TensorFlow, Theano, Caffe and CNTK have a static view of the world. One has to build a neural network, and reuse the same structure again and again. Changing the way the network behaves means that one has to start from scratch. [… Pytorch] allows you to change the way your network behaves arbitrarily with zero lag or overhead.

Of course the overhead is not truly zero; rather they have shifted the overhead baseline down a little. But whatever, it’s comparatively convenient.

Not, however, as convenient for my purposes as frameworks that avoid industrial neural frameworks altogether when what I usually want is fairly basic autodiff.

jax is a python autodiff jit that I currently use, and also julia has many many options some of which I use, and overall this is much more efficient for me than either .

Getting started

DSP in pytorch

Keunwoo Choi, Inverse STFT, Harmonic Percussive separation.

Custom functions

There is some bad advice in the manual

nn exports two kinds of interfaces – modules and their functional versions. You can extend it in both ways, but we recommend using modules for all kinds of layers, that hold any parameters or buffers, and recommend using a functional form parameter-less operations like activation functions, pooling, etc.

So, important missing information.

It’s just as well it’s easy to roll your own recurrent nets because the default implementations are bad

The default RNN layer is heavily optimised using CUDNN, which is sweet, but you only have a choice of 2 activation functions, and neither of them is “linear”.

Ding Ke made a beautiful and simple RNN implementation.

Logging and visualizing training

Visualising graphs

Fiddly. The official way is via ONNX.

conda install -c ezyang onnx pydot # or
pip install onnx pydot
brew cask install netron # or
pip install netron
brew install graphviz

Also available, pytorchviz.

pip install git+
from pytorchviz import make_dot
y = model(x)
make_dot(y, params = dict(model.named_parameters())

Utility libraries, derived software

Pytorch ships with a lot of included functionality, so you don’t necessarily ned to wrap it in anything else. Nonetheless, you can for specific use. NB this is list not up to date.


Like other deep learning frameworks, there is some basic NLP support in pytorch; see pytorch.text.

flair is a commercially-backed NLP framework.


Pump graphs to a visualisation server. No pytorch-specific, but seems well-integrated. visdom


pytorch + Bayes = pyro, an Edwardlib competitor.

Pyro launch announcment:

We believe the critical ideas to solve AI will come from a joint effort among a worldwide community of people pursuing diverse approaches. By open sourcing Pyro, we hope to encourage the scientific world to collaborate on making AI tools more flexible, open, and easy-to-use. We expect the current (alpha!) version of Pyro will be of most interest to probabilistic modelers who want to leverage large data sets and deep networks, PyTorch users who want easy-to-use Bayesian computation, and data scientists ready to explore the ragged edge of new technology.


inferno is a grab-bag library for torch.

Current features include:

I’m not sold on this one; A whole new library to reduce an already small amount of boilerplate, without adding any new non-trivial capabilities.


TNT is a reimplementation of some lua library that lua torch users used, that the current generation of ML users never witnessed. I think it aims to be semi-official library for pytorch, but it’s not especially active.

TNT (imported as torchnet) is a framework for PyTorch which provides a set of abstractions for PyTorch aiming at encouraging code re-use as well as encouraging modular programming. It provides powerful dataloading, logging, and visualization utilities.[…]

For example, TNT provides simple methods to record model performance in the torchnet.meter module and to log them to Visdom (or in the future, TensorboardX) with the torchnet.logging.

TNT docs.


Memory leaks

Apparently you use normal python garbage collector analysis.

import torch
import gc
for obj in gc.get_objects():
        if torch.is_tensor(obj) or (hasattr(obj, 'data') and torch.is_tensor(
            print(type(obj), obj.size())
    except Exception as e:

See also usual python debugging.