amazon-braket / BraketAHS.jl

Simulating Rydberg atom dynamics with Tensor Networks
Apache License 2.0
20 stars 6 forks source link

Support for ITensorGPU backend #15

Open ykharkov opened 5 months ago

ykharkov commented 5 months ago

In the current version the BraketAHS.jl package supports only CPU-based computation.

Goal: Add support for the GPU computations by leveraging ITensorsGPU.jl package.

User should have and ability to select the computational backend in mps_runner.jl. The GPU acceleration should be enabled for the TEBD evolution loop (and for TDVP once it is implemented).

Options that User can choose from: [CPU (default), GPU]. These options are supplied as a command line argument: --backend CPU [GPU].