Keras is primarily designed to work with TensorFlow. However, for integrating Keras-like functionality in PyTorch, you can explore high-level interfaces like PyTorch Lightning, which simplifies the PyTorch code for more straightforward model development.