https://pytorch.org/tutorials/intermediate/dynamic_quantization_bert_tutorial.html