Use tensorflow.keras module on Summit

[j-woz]

Not sure how else to handle this

[j-woz]

module ibm-wml requires a different module naming scheme