Can BERT router source code (esp. for training) be added to the repo?

@iojw can BERT router source code for training be added to the repo? Thanks.