WarOS - Quantum Computing & Post-Quantum Cryptography for Python

High-performance quantum circuit simulation and post-quantum cryptography, powered by Rust.

Install

pip install waros

Quick Start

import waros

circuit = waros.Circuit(2)
circuit.h(0)
circuit.cnot(0, 1)
circuit.measure_all()

simulator = waros.Simulator(seed=42)
result = simulator.run(circuit, shots=10_000)
print(result)
print(circuit.stats())
print(circuit.draw())

Realistic Noise

noise = waros.NoiseModel.ibm()
simulator = waros.Simulator(seed=42, noise=noise)
result = simulator.run(circuit, shots=10_000)
result.histogram()

Post-Quantum Cryptography

from waros import crypto

pk, sk = crypto.kem_keygen()
ct, shared_secret_a = crypto.kem_encapsulate(pk)
shared_secret_b = crypto.kem_decapsulate(sk, ct)
assert shared_secret_a == shared_secret_b

pk, sk = crypto.sign_keygen()
signature = crypto.sign(sk, b"Hello WarOS")
assert crypto.verify(pk, b"Hello WarOS", signature)

digest = crypto.sha3_256(b"WarOS")
random_data = crypto.random_bytes(32)

OpenQASM Support

qasm_source = """
OPENQASM 2.0;
include "qelib1.inc";
qreg q[2];
creg c[2];
h q[0];
cx q[0], q[1];
measure q[0] -> c[0];
measure q[1] -> c[1];
"""

circuit = waros.parse_qasm(qasm_source)
result = waros.Simulator(seed=42).run(circuit, shots=1_000)

Notebook and Compatibility Helpers

from waros.compat import QuantumCircuit

qc = QuantumCircuit(2, 2)
qc.h(0)
qc.cx(0, 1)
qc.measure([0, 1], [0, 1])

result = qc.run(shots=1_000)
display(result)

One-Line Algorithm Demos

bell = waros.algorithms.run_bell_state(shots=1_000, seed=42)
grover = waros.algorithms.run_grover(shots=1_000, seed=42)
teleport = waros.algorithms.run_teleport(state_theta=1.047, shots=1_000, seed=42)
factors = waros.algorithms.shor_factor(15, seed=42)
energy = waros.algorithms.vqe_hydrogen(max_iterations=30, seed=42)

Part of the WarOS hybrid quantum-classical operating system project.