XLA GSPMD and Keras

77 views

Skip to first unread message

unread,

May 19, 2022, 6:59:03 AM5/19/22

to XLA development

Hi all,

is Keras compatible with XLA GSPMD sharding engine?

Is there any example for using GSPMD on a Keras generated graph, for training?

Thanks,

Moshe

unread,

Sep 16, 2022, 3:05:47 PM9/16/22

to XLA development

Hi,

Yes, it is.

The sharding APIs described here lower to GSPMD.

Reply all

Reply to author

Forward

0 new messages