Как реализовать CRF в тензор потока 2

Question

Я хотел бы следовать этому руководству: https://www.kaggle.com/nikkisharma536/ner-with-bilstm-and-crf

, но использовать Tensorflow 2. Требуется, чтобы я внедрил CRF.

Как это сделать в Tensorflow 2? Я знаю, что в tensorflow-addons есть класс CRF, но нет руководства по его использованию.