Icon for KerasIcon for PyTorch

Keras integrates with PyTorch

Integrates with

Keras 3 integrates with PyTorch as one of its three supported backends. When using the PyTorch backend, Keras models use PyTorch tensors and autograd, enabling access to the PyTorch ecosystem including torch.compile, PyTorch Lightning, and native PyTorch deployment options.