Add Callback support to when training remotely keras model

[yanndupis]

Description

We'd like to add Callback support so we can custom training. Currently when passing to fit, it doesn't work because we don't have generic serde support for callbacks.

Objectives/Key Results

• Train Keras model with callback passed to .fit.