Hey @chMoussa thank you for this. Two points that puzzle me: 1) wouldn't be easier to have a DensityMatrix type to be queried for instead of passing an argument all around ? It might not be idiomatic to Jax though. 2) Is there a need for NoiseProtocol to be an empty tuple ? Other than that, minor quibbles around type updates (Tuple -> tuple) and few other points.

Hey @RolandMacDoland,
For 1) I tried originally to have this class but inheriting the Array is not allowed, so that's why I decided to carry a boolean instead.
For 2) I thought we would avoid the need to use None by simply using an empty tuple.

Hey @chMoussa thank you for this. Two points that puzzle me: 1) wouldn't be easier to have a DensityMatrix type to be queried for instead of passing an argument all around ? It might not be idiomatic to Jax though. 2) Is there a need for NoiseProtocol to be an empty tuple ? Other than that, minor quibbles around type updates (Tuple -> tuple) and few other points.

Hey @RolandMacDoland, For 1) I tried originally to have this class but inheriting the Array is not allowed, so that's why I decided to carry a boolean instead. For 2) I thought we would avoid the need to use None by simply using an empty tuple.

Hey @RolandMacDoland , I managed to make it with a DensityMatrix class and a few singledispatch mathods to handle state vectors and density matrices differently.