https://forecastegy.com/posts/tw-bert-end-to-end-query-term-weighting-pytorch/
PyTorch Implementation of Google's TW-BERT for Information Retrieval