knead currently outputs .pb files, which are file formats very-well suited for deep learning. However, the standard way to deserialize these files in PyTorch is to create a Dataset class. This ought to be provided by knead.

Luckily, the code is already written: https://github.com/ccurro/font-bakers/blob/master/src/serialization/fontDataset.py#L9