Quantum Reinforcement Learning library: environments, policies and training loops with PennyLane.
Project description
qrl-qai: The quantum analogue of OpenAI's gym python framework
qrl-qai is a python framework built on top of Gymnasium, PennyLane, and PyTorch to serve as a central platform for everything quantum reinforcement learning.
The current release (0.3.0) offers 5 native quantum RL style environments:
- BlochSphereV0
- CompilerV0
- ErrorChannelV0
- ExpressibilityV0
- ProbabilityV0
Check out the Quickstart guide to get started.
Additionally, each significant release has an asssociated Google Colab and Lightning AI Studio for a hassle free experience. These are especially useful for users who want to quickly test out the environments without going through the installation process.
Lightnining AI Studio contains a Streamlit playground for no-code experimentation with the environments!
| Version | Google Colab | Lightning AI Studio | Documentation |
|---|---|---|---|
| 0.1.0 | - | - | Read The Docs |
| 0.2.0 | Read The Docs | ||
| 0.3.0 | - | - | Read The Docs |
📑 Table of Contents
🗺 Roadmap
🚀 Installation
Install the package directly from PyPI:
pip install qrl-qai
To save the episodes as mp4 videos, it is essential to have ffmpeg installed on your system. In the absence of ffmpeg, you can run the environments but save the episodes as gif only and not mp4.
To check if you have ffmpeg installed (Windows/Linux/Mac), you can try:
ffmpeg -version
To install ffmpeg, you can follow these steps:
Using Conda
You can also install FFmpeg within a conda environment:
conda install -c conda-forge ffmpeg
Windows
Option 1: Using Chocolatey (Recommended)
If you have Chocolatey installed:
choco install ffmpeg
✅ Option 2: Manual installation
-
Go to the official FFmpeg site → Click Windows → gyan.dev builds or BtbN builds.
-
Download the latest release full build ZIP.
-
Extract it (e.g., to C:\ffmpeg).
-
Add the bin folder to your PATH:
4.1 Press Win + R → sysdm.cpl → Advanced → Environment Variables
4.2 Edit Path → Add new entry: C:\ffmpeg\bin
Linux
Debian / Ubuntu:
sudo apt update
sudo apt install ffmpeg -y
Fedora:
sudo dnf install ffmpeg -y
Arch Linux:
sudo pacman -S ffmpeg
macOS
Using Homebrew:
brew install ffmpeg
🤝 Contributing
See our CONTRIBUTING.md for guidelines.
📜 License
This project is licensed under the Apache 2.0 License
📬 Contact
Initiated by Jay Shah
Email: jay.shah@qrlqai.com
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file qrl_qai-1.0.0.tar.gz.
File metadata
- Download URL: qrl_qai-1.0.0.tar.gz
- Upload date:
- Size: 50.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
328b676af69f18ca36b6ae351ffba4cebbb2208b26ab2631f40e4088bb4f0883
|
|
| MD5 |
70dbd281b454f83a31d603ffeb0c3041
|
|
| BLAKE2b-256 |
a5f30971f03714a7b692c47e9355b49b6227fbdcf8853152472df25ba5837181
|
File details
Details for the file qrl_qai-1.0.0-py3-none-any.whl.
File metadata
- Download URL: qrl_qai-1.0.0-py3-none-any.whl
- Upload date:
- Size: 50.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
5e4d7e46b85009e854a4d871074e634fc0f85925ca2d90dc7f6dbe0887ea541b
|
|
| MD5 |
a4d20313b585db78a204fd65e7cc46f4
|
|
| BLAKE2b-256 |
9c624730ba42057a184ab1820edc595ffed931170f282f13816673498158f0a8
|