Fixed kraus function and added tests

CQCL / qujax

Simulating quantum circuits with JAX

https://cqcl.github.io/qujax/

Apache License 2.0

58 stars 10 forks source link

Fixed kraus function and added tests #40

Closed SamDuffield closed 2 years ago

SamDuffield commented 2 years ago

Fixed the (multiple) kraus function, which previously threw an error.

Updated get_params_to_densitytensor_func to be able to take generic Kraus operators rather than only single unitary arrays as elements of the gate_seq argument