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 #23

Open potatoboiler opened 3 months ago

potatoboiler commented 3 months ago

Issue #, if available: https://github.com/amazon-braket/BraketAHS.jl/issues/15

Description of changes: Add CUDA backend (don't have Intel or Metal GPU so can't touch those yet). New to Julia so code may be un-idiomatic. Code currently has issues with normalizing the state vector \psi, not sure how to fix this (I am not familiar with the underlying science). Error message below:

[ Info: Sampling from final MPS state
ERROR: LoadError: sample: MPS is not normalized, norm=0.99999994

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.