torch transformers tensorflow numpy matplotlib Pillow