Failed to load image Python extension: Could not find module

I solved it by installing libpng and libjpeg with Homebrew and manually adding symlinks to /usr/local/lib.

I still think we should address this with a ticket since PyTorch packages are supposed to come with all dependencies included.