This repository contains some jupyter notebooks that train an RL (policy gradient) agent to prepare quantum states using JAX, and visualize the learned solution on the Bloch sphere using a QuTip function.
Required packages:
- jax=0.4.14, jaxlib=0.4.14
- numpy=1.25.2
- scipy=1.11.1
- matplotlib=3.7.1
- jupyterlab=4.2.3
- ipywidgets=8.1.3
- ipython=8.14.0