Quantum-ToT

Model Details

Quantum-ToT is a fine-tuned variant of Qwen3-1.7B, optimized for Chain-of-Thought (CoT) reasoning in quantum mechanics and quantum computing contexts. This model was trained using the moremilk/CoT_Reasoning_Quantum_Physics_And_Computing dataset — a curated collection of question–answer pairs that go beyond surface-level definitions to show the logical reasoning process behind quantum concepts.

The goal of this fine-tuning is to enhance the model’s ability to:

• Explain quantum principles with structured, step-by-step logic
• Reason through conceptual problems in quantum physics and computing
• Support educational and research applications that require interpretable reasoning chains

Uses

Direct Use

• Educational assistance in quantum physics and quantum computing
• AI tutors or reasoning assistants for STEM learning
• Conceptual reasoning benchmarks involving quantum phenomena
• Research in reasoning-aware model behavior and CoT interpretability

Out of Scope

• Predicting new or unverified physical phenomena
• Running quantum simulations or algorithmic derivations
• Hardware-level quantum design
• Real-time physics predictions

Bias, Risks, and Limitations

• May hallucinate if prompted outside the quantum domain
• Not suitable for advanced quantum algorithm design or experimental predictions

How to Get Started with the Model

Use the code below to get started with the model.

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("khazarai/Quantum-ToT")
model = AutoModelForCausalLM.from_pretrained(
    "khazarai/Quantum-ToT",
    device_map={"": 0}
)

question = """
Explain the Heisenberg Uncertainty Principle in detail, including its mathematical formulation, physical implications, and common misconceptions.
"""

messages = [
    {"role" : "user", "content" : question}
]
text = tokenizer.apply_chat_template(
    messages,
    tokenize = False,
    add_generation_prompt = True,
    enable_thinking = True,
)

from transformers import TextStreamer
_ = model.generate(
    **tokenizer(text, return_tensors = "pt").to("cuda"),
    max_new_tokens = 3000, 
    temperature = 0.6,
    top_p = 0.95,
    top_k = 20,
    streamer = TextStreamer(tokenizer, skip_prompt = True),
)

Dataset

Dataset: moremilk/CoT_Reasoning_Quantum_Physics_And_Computing

This dataset contains rich reasoning-based question–answer pairs covering:

• Core quantum principles: superposition, entanglement, measurement
• Effects of quantum gates (Hadamard, Pauli-X/Y/Z, etc.) on qubits
• Multi-qubit reasoning (e.g., Bell states, entangled systems)
• Basic quantum algorithms and logical operations
• Probabilistic interpretation of measurement outcomes

Each entry includes:

• think block → model’s internal reasoning process
• answer block → final concise explanation or solution

The dataset focuses on conceptual understanding rather than heavy mathematical derivations or complex quantum hardware design.