[adamisntdead]

I wrote something similar a couple of years ago - https://github.com/adamisntdead/QuSimPy

Happy to answer any questions people have, including on other simulation methods other than state vector!

[reikonomusha]

One of the aspects emphasized in TFA is being able to simulate gates of any dimension/number of qubits (like a 3-qubit Toffoli or a 5-qubit Molmer-Sorensen), instead of just 1- and 2-qubit gates. Have you thought about extending your simulator to support gates of greater than two qubits?

[adamisntdead]

I guess back when I wrote it I didn't see the need given that single qubit gates along with CNOT are universal and they're implemented. I think airing on the side of 'as few features as is educational' was my mentality on this.

[JBits]

I know about the conventional 2-qubit Molmer-Sorensen but not the 5-qubit version. Could I ask what it is?

[reikonomusha]

See e.g. equations (5) and (6) from [1], where any qubit subset of a system might undergo the action of the M–S gate. But definitely the usual presentation of the physics of M–S is on a two qubit system.

[1] https://arxiv.org/abs/1601.06819