amazon-braket / amazon-braket/BraketAHS.jl
Support for ITensorGPU backend
Open
enhancement
good first issue
- Dominant language
- Julia
- Stars
- 24
- Forks
- 7
- PR merge metrics
- No merged PRs in 30d
Description
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].
Contributor guide
Assessment
This issue has not been assessed yet.