Grad-Cam Explainer for RLlib

How severe does this issue affect your experience of using Ray?

  • Medium: It contributes to significant difficulty to complete my task, but I can work around it.

Hi all,
I would like to add a Grad-Cam explainer to my PPO policy network. Assume my Grad-cam class is ready. Does anyone know how to incorporate it into the RLlib while computing a single action for a trained agent?
Thanks!