Save and load trained keras-gp model

Dear Maruan,
I was wondering if there is a way to save and load a trained keras-gp model.
In keras this is usually done by using:

model.save('my_model.h5')
model = load_model('my_model.h5')

Following this way, it results in an error not recognising the GP layer.
Is there a way to correctly save and load a keras-gp model?

I would appreciate any hint!
Best,
Marc