Reply to InvalidArgumentError: Cannot assign a device for operation model/bert_block/encoder_0/multiheadattention/query/einsum/Einsum/ReadVariableOp Hello there ! I am running into the same error with the ReadVariableOp using the Adam optimizer. Changing to Adam solved the issue. Do we have an update on the above ? Machine Learning & AI General Mar ’23