├── .github ├── dependabot.yml └── workflows │ ├── build_and_test.yml │ ├── docs.yml │ ├── docs │ └── requirements.txt │ └── lint.yml ├── .gitignore ├── .pylintrc ├── CITATION.cff ├── LICENSE ├── README.md ├── docs ├── _static │ └── css │ │ └── custom.css ├── conf.py ├── densitytensor.rst ├── densitytensor │ ├── all_zeros_densitytensor.rst │ ├── densitytensor_to_measured_densitytensor.rst │ ├── densitytensor_to_measurement_probabilities.rst │ ├── get_densitytensor_to_expectation_func.rst │ ├── get_densitytensor_to_sampled_expectation_func.rst │ ├── get_params_to_densitytensor_func.rst │ ├── kraus.rst │ ├── partial_trace.rst │ └── statetensor_to_densitytensor.rst ├── examples.md ├── gates.rst ├── getting_started.rst ├── index.rst ├── logo.svg ├── logo_dark_mode.svg ├── requirements.txt ├── statetensor.rst ├── statetensor │ ├── all_zeros_statetensor.rst │ ├── apply_gate.rst │ ├── get_params_to_statetensor_func.rst │ ├── get_params_to_unitarytensor_func.rst │ ├── get_statetensor_to_expectation_func.rst │ └── get_statetensor_to_sampled_expectation_func.rst ├── utils.rst └── utils │ ├── bitstrings_to_integers.rst │ ├── check_circuit.rst │ ├── integers_to_bitstrings.rst │ ├── print_circuit.rst │ ├── repeat_circuit.rst │ ├── sample_bitstrings.rst │ └── sample_integers.rst ├── examples ├── barren_plateaus.ipynb ├── classification.ipynb ├── generative_modelling.ipynb ├── heisenberg_vqe.ipynb ├── hidalgo_stamp.txt ├── maxcut_vqe.ipynb ├── noise_channel.ipynb ├── qaoa.ipynb ├── reducing_jit_compilation_time.ipynb └── variational_inference.ipynb ├── qujax ├── __init__.py ├── densitytensor.py ├── densitytensor_observable.py ├── gates.py ├── statetensor.py ├── statetensor_observable.py ├── typing.py ├── utils.py └── version.py ├── setup.py └── tests ├── test_densitytensor.py ├── test_expectations.py ├── test_gates.py ├── test_statetensor.py └── test_utils.py /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/build_and_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.github/workflows/build_and_test.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.github/workflows/docs/requirements.txt -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/.pylintrc -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/README.md -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/densitytensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor.rst -------------------------------------------------------------------------------- /docs/densitytensor/all_zeros_densitytensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/all_zeros_densitytensor.rst -------------------------------------------------------------------------------- /docs/densitytensor/densitytensor_to_measured_densitytensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/densitytensor_to_measured_densitytensor.rst -------------------------------------------------------------------------------- /docs/densitytensor/densitytensor_to_measurement_probabilities.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/densitytensor_to_measurement_probabilities.rst -------------------------------------------------------------------------------- /docs/densitytensor/get_densitytensor_to_expectation_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/get_densitytensor_to_expectation_func.rst -------------------------------------------------------------------------------- /docs/densitytensor/get_densitytensor_to_sampled_expectation_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/get_densitytensor_to_sampled_expectation_func.rst -------------------------------------------------------------------------------- /docs/densitytensor/get_params_to_densitytensor_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/get_params_to_densitytensor_func.rst -------------------------------------------------------------------------------- /docs/densitytensor/kraus.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/kraus.rst -------------------------------------------------------------------------------- /docs/densitytensor/partial_trace.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/partial_trace.rst -------------------------------------------------------------------------------- /docs/densitytensor/statetensor_to_densitytensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/densitytensor/statetensor_to_densitytensor.rst -------------------------------------------------------------------------------- /docs/examples.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/examples.md -------------------------------------------------------------------------------- /docs/gates.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/gates.rst -------------------------------------------------------------------------------- /docs/getting_started.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/getting_started.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/logo.svg -------------------------------------------------------------------------------- /docs/logo_dark_mode.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/logo_dark_mode.svg -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/statetensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor.rst -------------------------------------------------------------------------------- /docs/statetensor/all_zeros_statetensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor/all_zeros_statetensor.rst -------------------------------------------------------------------------------- /docs/statetensor/apply_gate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor/apply_gate.rst -------------------------------------------------------------------------------- /docs/statetensor/get_params_to_statetensor_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor/get_params_to_statetensor_func.rst -------------------------------------------------------------------------------- /docs/statetensor/get_params_to_unitarytensor_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor/get_params_to_unitarytensor_func.rst -------------------------------------------------------------------------------- /docs/statetensor/get_statetensor_to_expectation_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor/get_statetensor_to_expectation_func.rst -------------------------------------------------------------------------------- /docs/statetensor/get_statetensor_to_sampled_expectation_func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/statetensor/get_statetensor_to_sampled_expectation_func.rst -------------------------------------------------------------------------------- /docs/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils.rst -------------------------------------------------------------------------------- /docs/utils/bitstrings_to_integers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/bitstrings_to_integers.rst -------------------------------------------------------------------------------- /docs/utils/check_circuit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/check_circuit.rst -------------------------------------------------------------------------------- /docs/utils/integers_to_bitstrings.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/integers_to_bitstrings.rst -------------------------------------------------------------------------------- /docs/utils/print_circuit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/print_circuit.rst -------------------------------------------------------------------------------- /docs/utils/repeat_circuit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/repeat_circuit.rst -------------------------------------------------------------------------------- /docs/utils/sample_bitstrings.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/sample_bitstrings.rst -------------------------------------------------------------------------------- /docs/utils/sample_integers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/docs/utils/sample_integers.rst -------------------------------------------------------------------------------- /examples/barren_plateaus.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/barren_plateaus.ipynb -------------------------------------------------------------------------------- /examples/classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/classification.ipynb -------------------------------------------------------------------------------- /examples/generative_modelling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/generative_modelling.ipynb -------------------------------------------------------------------------------- /examples/heisenberg_vqe.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/heisenberg_vqe.ipynb -------------------------------------------------------------------------------- /examples/hidalgo_stamp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/hidalgo_stamp.txt -------------------------------------------------------------------------------- /examples/maxcut_vqe.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/maxcut_vqe.ipynb -------------------------------------------------------------------------------- /examples/noise_channel.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/noise_channel.ipynb -------------------------------------------------------------------------------- /examples/qaoa.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/qaoa.ipynb -------------------------------------------------------------------------------- /examples/reducing_jit_compilation_time.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/reducing_jit_compilation_time.ipynb -------------------------------------------------------------------------------- /examples/variational_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/examples/variational_inference.ipynb -------------------------------------------------------------------------------- /qujax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/__init__.py -------------------------------------------------------------------------------- /qujax/densitytensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/densitytensor.py -------------------------------------------------------------------------------- /qujax/densitytensor_observable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/densitytensor_observable.py -------------------------------------------------------------------------------- /qujax/gates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/gates.py -------------------------------------------------------------------------------- /qujax/statetensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/statetensor.py -------------------------------------------------------------------------------- /qujax/statetensor_observable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/statetensor_observable.py -------------------------------------------------------------------------------- /qujax/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/typing.py -------------------------------------------------------------------------------- /qujax/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/qujax/utils.py -------------------------------------------------------------------------------- /qujax/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "1.1.0" 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_densitytensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/tests/test_densitytensor.py -------------------------------------------------------------------------------- /tests/test_expectations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/tests/test_expectations.py -------------------------------------------------------------------------------- /tests/test_gates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/tests/test_gates.py -------------------------------------------------------------------------------- /tests/test_statetensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/tests/test_statetensor.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Quantinuum/qujax/HEAD/tests/test_utils.py --------------------------------------------------------------------------------