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.