Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/pennylaneai/catalyst
A JIT compiler for hybrid quantum programs in PennyLane
https://github.com/pennylaneai/catalyst
autodiff automatic-differentiation jax jit llvm mlir pennylane python qir quantum quantum-compiler quantum-computing
Last synced: 5 days ago
JSON representation
A JIT compiler for hybrid quantum programs in PennyLane
- Host: GitHub
- URL: https://github.com/pennylaneai/catalyst
- Owner: PennyLaneAI
- License: apache-2.0
- Created: 2023-02-17T02:23:20.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2025-01-15T03:41:59.000Z (8 days ago)
- Last Synced: 2025-01-15T05:54:16.779Z (8 days ago)
- Topics: autodiff, automatic-differentiation, jax, jit, llvm, mlir, pennylane, python, qir, quantum, quantum-compiler, quantum-computing
- Language: Python
- Homepage: https://docs.pennylane.ai/projects/catalyst
- Size: 13.3 MB
- Stars: 148
- Watchers: 14
- Forks: 40
- Open Issues: 147
-
Metadata Files:
- Readme: README.md
- Contributing: .github/CONTRIBUTING.md
- License: LICENSE
- Code of conduct: .github/CODE_OF_CONDUCT.md
- Citation: CITATION.cff
Awesome Lists containing this project
README
[](https://github.com/PennyLaneAI/catalyst/actions/workflows/check-catalyst.yaml)
[](https://codecov.io/gh/PennyLaneAI/catalyst)
[](https://docs.pennylane.ai/projects/catalyst)
[](https://doi.org/10.21105/joss.06720)
[](https://pypi.org/project/PennyLane-Catalyst)
[](https://discuss.pennylane.ai)
[](https://www.apache.org/licenses/LICENSE-2.0)
Catalyst is an experimental package that enables just-in-time (JIT) compilation of hybrid
quantum-classical programs.
**Catalyst is currently under heavy development — if you have suggestions on the API or use-cases
you'd like to be covered, please open an GitHub issue or reach out. We'd love to hear about how
you're using the library, collaborate on development, or integrate additional devices and
frontends.**
## Key Features
- Compile the entire quantum-classical workflow, including any optimization loops.
- Use Catalyst alongside PennyLane directly from Python. Simply decorate quantum code and hybrid
functions with `@qjit`, leading to significant performance improvements over standard Python
execution.
- Access advanced control flow that supports both quantum and classical instructions.
- Infrastructure for both quantum *and* classical compilation, allowing you to compile quantum
circuits that contain control flow.
- Built to be end-to-end differentiable.
- Support for [PennyLane-Lightning](https://github.com/PennyLaneAI/pennylane-lightning) high performance
simulators, and [Amazon Braket](https://amazon-braket-pennylane-plugin-python.readthedocs.io)
devices. Additional hardware support, including QPUs, to come.
## Overview
Catalyst currently consists of the following components:
- [Catalyst Compiler](https://docs.pennylane.ai/projects/catalyst/en/latest/modules/mlir.html).
The core Catalyst compiler is built using [MLIR](https://mlir.llvm.org/), with the addition of a
quantum dialect used to represent quantum instructions. This allows for a high-level intermediate
representation of the classical and quantum components of the program, resulting in advantages
during optimization. Once optimized, the compiler lowers the representation down to LLVM +
[QIR](https://www.qir-alliance.org/), and a machine binary is produced.
- [Catalyst
Runtime](https://docs.pennylane.ai/projects/catalyst/en/latest/modules/runtime.html).
The runtime is a C++ runtime with multiple-device support based on QIR that enables the execution
of Catalyst-compiled quantum programs. A complete list of all backend devices along with the quantum
instruction set supported by these runtime implementations can be found by visiting
[the runtime documentation](https://docs.pennylane.ai/projects/catalyst/en/latest/modules/runtime.html).
In addition, we also provide a Python frontend for [PennyLane](https://pennylane.ai) and [JAX](https://jax.readthedocs.io):
- [PennyLane JAX frontend](https://docs.pennylane.ai/projects/catalyst/en/latest/modules/frontend.html).
A Python library that provides a `@qjit` decorator to just-in-time compile PennyLane hybrid
quantum-classical programs. In addition, the frontend package provides Python functions for
defining Catalyst-compatible control flow structures, gradient, and mid-circuit measurement.
## Installation
Catalyst is officially supported on Linux (x86_64, aarch64) and macOS (arm64, x86_64) platforms,
and pre-built binaries are being distributed via the Python Package Index (PyPI) for Python versions 3.10 and
higher. To install it, simply run the following ``pip`` command:
```console
pip install pennylane-catalyst
```
If you wish to contribute to Catalyst or develop against our runtime or compiler, instructions for
[building from source](https://docs.pennylane.ai/projects/catalyst/en/latest/dev/installation.html#minimal-building-from-source-guide)
are also available.
## Trying Catalyst with PennyLane
To get started using the Catalyst JIT compiler from Python, check out our [quick start
guide](https://docs.pennylane.ai/projects/catalyst/en/latest/dev/quick_start.html), as well as our
various examples and tutorials in our [documentation](https://docs.pennylane.ai/projects/catalyst).
For an introduction to quantum computing and quantum machine learning, you can also visit the
PennyLane website for [tutorials, videos, and demonstrations](https://pennylane.ai/qml).
## Roadmap
- **Frontend:** As we continue to build out Catalyst, the PennyLane frontend will likely be
upstreamed into PennyLane proper, providing native JIT functionality built-in to PennyLane. The
Catalyst compiler and runtime will remain part of the Catalyst project. *If you are interested in
working on additional frontends for Catalyst, please get in touch.*
- **Compiler:** We will continue to build out the compiler stack, and add quantum compilation
routines. This includes an API for providing or writing Catalyst-compatible compilation routines.
In addition, we will be improving the autodifferentiation support, and adding support for
classical autodiff, additional quantum gradients, and quantum-aware optimization methods.
- **Runtime:** We will be adding support for more devices, including quantum hardware devices. In
addition, we will be building out support for hetereogeneous execution. *If you are interested in
working on connecting a quantum device with Catalyst, please get in touch.*
To get the details right, we need your help — please send us your use cases by starting a
conversation, or trying Catalyst out.
## Contributing to Catalyst
We welcome contributions — simply fork the Catalyst repository, and then make a [pull
request](https://help.github.com/articles/about-pull-requests/) containing your contribution.
We also encourage bug reports, suggestions for new features and enhancements.
## Support
- **Source Code:** https://github.com/PennyLaneAI/catalyst
- **Issue Tracker:** https://github.com/PennyLaneAI/catalyst/issues
If you are having issues, please let us know by posting the issue on our GitHub issue tracker.
We also have a [PennyLane discussion forum](https://discuss.pennylane.ai)—come join the community
and chat with the PennyLane team.
Note that we are committed to providing a friendly, safe, and welcoming environment for all. Please
read and respect the [Code of Conduct](.github/CODE_OF_CONDUCT.md).
## Authors
Catalyst is the work of [many contributors](https://github.com/PennyLaneAI/catalyst/graphs/contributors).
If you are doing research using Catalyst, please cite our paper:
```bibtex
@article{
Ittah2024,
doi = {10.21105/joss.06720},
url = {https://doi.org/10.21105/joss.06720},
year = {2024},
publisher = {The Open Journal},
volume = {9},
number = {99},
pages = {6720},
author = {David Ittah and Ali Asadi and Erick Ochoa Lopez and Sergei Mironov and Samuel Banning and Romain Moyard and Mai Jacob Peng and Josh Izaac},
title = {Catalyst: a Python JIT compiler for auto-differentiable hybrid quantum programs},
journal = {Journal of Open Source Software}
}
```
## License
Catalyst is **free** and **open source**, released under the Apache License, Version 2.0.
## Acknowledgements
Catalyst makes use of the following libraries and tools, which are under their own respective
licenses:
- [JAX](https://github.com/jax-ml/jax)
- [TensorFlow](https://github.com/tensorflow/tensorflow)
- [OpenXLA](https://github.com/openxla/xla)
- [LLVM/MLIR](https://github.com/llvm/llvm-project)
- [EnzymeAD](https://github.com/EnzymeAD/Enzyme)
- [pybind11](https://github.com/pybind/pybind11)
- [nanobind](https://github.com/wjakob/nanobind)
- [accelerate-lapacke](https://github.com/lepus2589/accelerate-lapacke)
- [LAPACK](https://github.com/Reference-LAPACK/lapack)
- [TOML++](https://github.com/marzer/tomlplusplus)