tyxonq 0.1.1__py3-none-any.whl

tyxonq/__init__.py

@@ -0,0 +1,70 @@
+__version__ = "0.1.1"
+__author__ = "TensorCircuit Authors"
+__creator__ = "refraction-ray"
+
+from .utils import gpu_memory_share
+
+gpu_memory_share()
+
+from .about import about
+from .cons import (
+    backend,
+    set_backend,
+    set_dtype,
+    set_contractor,
+    get_backend,
+    get_dtype,
+    get_contractor,
+    set_function_backend,
+    set_function_dtype,
+    set_function_contractor,
+    runtime_backend,
+    runtime_dtype,
+    runtime_contractor,
+)  # prerun of set hooks
+from . import gates
+from . import basecircuit
+from .gates import Gate
+from .circuit import Circuit, expectation
+from .mpscircuit import MPSCircuit
+from .densitymatrix import DMCircuit as DMCircuit_reference
+from .densitymatrix import DMCircuit2
+
+DMCircuit = DMCircuit2  # compatibility issue to still expose DMCircuit2
+from .gates import num_to_tensor, array_to_tensor
+from .vis import qir2tex, render_pdf
+from . import interfaces
+from . import templates
+from . import results
+from . import quantum
+from .quantum import QuOperator, QuVector, QuAdjointVector, QuScalar
+from . import compiler
+from . import cloud
+from . import fgs
+from .fgs import FGSSimulator
+
+try:
+    from . import keras
+    from .keras import KerasLayer, KerasHardwareLayer
+except ModuleNotFoundError:
+    pass  # in case tf is not installed
+
+try:
+    from . import torchnn
+    from .torchnn import TorchLayer, TorchHardwareLayer
+except ModuleNotFoundError:
+    pass  # in case torch is not installed
+
+try:
+    import qiskit
+
+    qiskit.QuantumCircuit.cnot = qiskit.QuantumCircuit.cx
+    qiskit.QuantumCircuit.toffoli = qiskit.QuantumCircuit.ccx
+    qiskit.QuantumCircuit.fredkin = qiskit.QuantumCircuit.cswap
+
+    # amazing qiskit 1.0 nonsense...
+except ModuleNotFoundError:
+    pass
+
+# just for fun
+from .asciiart import set_ascii

tyxonq/about.py

@@ -0,0 +1,119 @@
+"""
+Prints the information for tyxonq installation and environment.
+"""
+
+import platform
+import sys
+import numpy
+
+
+def about() -> None:
+    """
+    Prints the information for tyxonq installation and environment.
+    """
+    print(f"OS info: {platform.platform(aliased=True)}")
+    print(
+        f"Python version: {sys.version_info[0]}.{sys.version_info[1]}.{sys.version_info[2]}"
+    )
+    print(f"Numpy version: {numpy.__version__}")
+
+    try:
+        import scipy
+
+        print(f"Scipy version: {scipy.__version__}")
+    except ModuleNotFoundError:
+        print(f"Scipy is not installed")
+
+    try:
+        import pandas
+
+        print(f"Pandas version: {pandas.__version__}")
+    except ModuleNotFoundError:
+        print(f"Pandas is not installed")
+
+    try:
+        import tensornetwork as tn
+
+        print(f"TensorNetwork version: {tn.__version__}")
+    except ModuleNotFoundError:
+        print(f"TensorNetwork is not installed")
+
+    try:
+        import cotengra
+
+        try:
+            print(f"Cotengra version: {cotengra.__version__}")
+        except AttributeError:
+            print(f"Cotengra: installed")
+    except ModuleNotFoundError:
+        print(f"Cotengra is not installed")
+
+    try:
+        import tensorflow as tf
+
+        print(f"TensorFlow version: {tf.__version__}")
+        print(f"TensorFlow GPU: {tf.config.list_physical_devices('GPU')}")
+        print(f"TensorFlow CUDA infos: {dict(tf.sysconfig.get_build_info())}")
+    except ModuleNotFoundError:
+        print(f"TensorFlow is not installed")
+
+    try:
+        import jax
+
+        print(f"Jax version: {jax.__version__}")
+        try:
+            device = jax.devices("gpu")
+            print(f"Jax GPU: {device}")
+        except RuntimeError:
+            print(f"Jax installation doesn't support GPU")
+    except ModuleNotFoundError:
+        print(f"Jax is not installed")
+
+    try:
+        import jaxlib
+
+        print(f"JaxLib version: {jaxlib.__version__}")
+    except ModuleNotFoundError:
+        print(f"JaxLib is not installed")
+
+    try:
+        import torch
+
+        print(f"PyTorch version: {torch.__version__}")
+        print(f"PyTorch GPU support: {torch.cuda.is_available()}")
+        print(
+            f"PyTorch GPUs: {[torch.cuda.device(i) for i in range(torch.cuda.device_count())]}"
+        )
+        if torch.version.cuda is not None:
+            print(f"Pytorch cuda version: {torch.version.cuda}")
+    except ModuleNotFoundError:
+        print(f"PyTorch is not installed")
+
+    try:
+        import cupy
+
+        print(f"Cupy version: {cupy.__version__}")
+    except ModuleNotFoundError:
+        print(f"Cupy is not installed")
+
+    try:
+        import qiskit
+
+        print(f"Qiskit version: {qiskit.__version__}")
+    except ModuleNotFoundError:
+        print(f"Qiskit is not installed")
+
+    try:
+        import cirq
+
+        print(f"Cirq version: {cirq.__version__}")
+    except ModuleNotFoundError:
+        print(f"Cirq is not installed")
+
+    from tyxonq import __version__
+
+    print(f"tyxonq version {__version__}")
+
+
+if __name__ == "__main__":
+    about()