A major difference between BERT and LDA kwx implementations is that there are no visualization methods for BERT. It would be good to add a pyLDAvis style visualization of topic words as well as a t-SNE visualization of topic similarities. These would be added to kwx.visuals.