modify mlm example to support all backends#1856

Mrutyunjay01 wants to merge 3 commits intokeras-team:masterkeras-team/keras-io:masterfrom

Mrutyunjay01:bert-mlm-portMrutyunjay01/keras-io:bert-mlm-portCopy head branch name to clipboard