What is the best way to import the BERT Language model into python (PyTorch)?