/steady_state_jax

Illustration of how to optimize non-equilibrium quantum systems using JAX.

Primary LanguageJupyter Notebook

Fully differentiable optimization protocols for non-equilibrium steady states

authors: Rodrigo A. Vargas-Hernandez, Ricky T. Q. Chen, Kenneth A. Jung, and Paul Brumer


GOAL:

Optimizing quantum non-equilibrium steady state systems using gradient-based methods power by automatic differentiation.

arxiv paper


Diagram of the methodology proposed here.

Diagram

For a V-system,

Diagram

We maximize the transport of energy to the reaction center (RC) using gradient descent,

Kenny_system_GA_loc_eff

All simulations were carried in JAX. We implemented the vector-Jacobian product to differentiate the steady state to incorporate it in the framework of automatic differentiation. For more details about our work we created a Google Collab, where we better exaplain the details of our work.

Foo

other systems: We also optimized a three level quantum heat transfer model using gradient based methods. For more details, check our arxiv paper.

Diagram


Prerequisites: JAX