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
Metadata
Release files for qrl-qai 1.0.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| qrl_qai-1.0.0.tar.gz | 50.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| qrl_qai-1.0.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 101.3 kB
Release files / qrl_qai-1.0.0.tar.gz
| Download URL | qrl_qai-1.0.0.tar.gz |
|---|---|
| Size | 50.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
328b676af69f18ca36b6ae351ffba4cebbb2208b26ab2631f40e4088bb4f0883
|
|
BLAKE2b-256 checksum How to use checksums |
a5f30971f03714a7b692c47e9355b49b6227fbdcf8853152472df25ba5837181
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.12.3
|
Release files / qrl_qai-1.0.0-py3-none-any.whl
| Download URL | qrl_qai-1.0.0-py3-none-any.whl |
|---|---|
| Size | 50.6 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
5e4d7e46b85009e854a4d871074e634fc0f85925ca2d90dc7f6dbe0887ea541b
|
|
BLAKE2b-256 checksum How to use checksums |
9c624730ba42057a184ab1820edc595ffed931170f282f13816673498158f0a8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.12.3
|