How to re-use the keras-bert implementation for question answering task?

Has anyone tried replicating this in keras with SQUAD dataset? It is not clear how can we prepare the input data for a custom bert model like having a LSTM on top of bert-base.