Fast Wave: Um pacote Python para um cálculo eficiente e preciso de funções de onda de estados de Fock

Abstract

The limitations in access to quantum hardware drive the search for efficient and accessible tools for quantum computing simulation. The quality of the simulation is determined by its efficiency and accuracy, while access can be facilitated by open source initiatives. In the context of photonic quantum computing, the representation of Fock states is fundamental, since the wave function of each state describes a well-defined quantity of photons. This work proposes a hybrid algorithm for the accurate and efficient calculation of the wave function of these states. The first part of the algorithm uses previously calculated coefficients stored in a matrix, while the second part employs a recursive formulation, optimizing performance and accuracy. It serves as the basis for the development of Fast Wave, a Python package made available to the community. The performance of the algorithm was improved in the package's modules, which use the Numba and Cython libraries. Comparisons between Fast Wave and the Mr Mustard package, developed by the company Xanadu, show that Fast Wave outperforms Mr Mustard in computational efficiency without compromising numerical reliability. Furthermore, part of the code from one of the Fast Wave modules was recently integrated into QuTip, one of the leading quantum mechanics packages in Python, consolidating its relevance in the scientific community.