lightning.kokkos

The lightning.kokkos device gives you portability; run on parallelized CPUs, NVIDIA GPUs, or AMD GPUs—whatever you have available!

Recommended for:

• Workflows on HPC-targeted hardware platforms.
• Fast gradients using the adjoint differentiation method.
• Distributed state vector support when on CPU using OpenMP.
• Linux operating systems.

---

Documentation

To learn more, please visit the device documentation:

• lightning.kokkos documentation

See all Lightning devices:

• lightning.gpu
• lightning.kokkos
• lightning.qubit
• lightning.tensor

---

Installation

For details on installation and dependencies, visit the lightning.kokkos installation page.

---

Device Initialization

Initialize the device in PennyLane with:

import pennylane as qml
dev = qml.device("lightning.kokkos", wires=2)

For more details on device settings and keyword arguments, see the device documentation.