QuForge: A Library for Qudits Simulation

TL;DR

QuForge is a Python library that enables efficient simulation of qudit-based quantum circuits, supporting differentiable programming and hardware acceleration for advanced quantum computing research.

Contribution

It introduces a flexible, GPU-accelerated library for simulating qudit quantum circuits with differentiable capabilities, expanding tools available for quantum machine learning.

Findings

Supports arbitrary qudit dimensions

Reduces memory usage with sparse operations

Enables quantum machine learning implementations

Abstract

Quantum computing with qudits, an extension of qubits to multiple levels, is a research field less mature than qubit-based quantum computing. However, qudits can offer some advantages over qubits, by representing information with fewer separated components. In this article, we present QuForge, a Python-based library designed to simulate quantum circuits with qudits. This library provides the necessary quantum gates for implementing quantum algorithms, tailored to any chosen qudit dimension. Built on top of differentiable frameworks, QuForge supports execution on accelerating devices such as GPUs and TPUs, significantly speeding up simulations. It also supports sparse operations, leading to a reduction in memory consumption compared to other libraries. Additionally, by constructing quantum circuits as differentiable graphs, QuForge facilitates the implementation of quantum machine learning algorithms, enhancing the capabilities and flexibility of quantum computing research.