Get BertSimple to run on Fox GPUs

[pmhalvor]

According to Andrey, the version of PyTorch I'm using is wrong. Needs to be the one built for nlpl project.

Checklist:

• [x] Load nlpl project
• [x] Check that nlpl modules are available
• [x] Load correct verison of PyTorch
• [x] Test model is correct cast to device