XLA GSPMD and Keras

Is Keras compatible with XLA GSPMD sharding engine?
Is there any example for using GSPMD on a Keras model, for training?

Same issue here and I can’t find any methods.

my finding is that GSPMD is not compatible w/ Keras.
however, the new DTensor mechanism is.
DTensor provides similar capabilities, which are not based on GSPMD engine.