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 on multiple CPU or GPU with Message Passing Interface (MPI).
- Linux operating systems.
Documentation
To learn more, please visit the device documentation:
See all Lightning devices:
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.