モデルだけkerasで作って、訓練はtfに投げるっていう手もある