As of 2023, keras 3.0.0 supports other types of backends, specifically torch and jax. It most likely would be a good idea to write tensorflow "agnostic" keras code in the future, since in a real world scenario there is some boilerplate data handling usually mixed with model creation and decoupling from tf might be useful.