keras-bert

Implementation of BERT that could load official pre-trained models for feature extraction and prediction