Amazon Interview Question

How BERT works; code simulation of discrete probability distribution