Are you using the same architecture for your TF/Keras model?
Could you post it just for the sake of debugging?
It looks like your model get’s stuck and just predicts constant logits.
Could you try to lower your learning rate a bit and see if something changes?