But Keras does not support PyTorch or MXNet. I think the design of Model, Block, and Layer like this is very intuitive and shared among several frameworks. I wish it could have multi-GPU/multi-node training capability (i.e. support horovod or gloo).