Please install below 2 in python environment before running the code:

Ollama: https://ollama.com/library/qwen

Gymnasium: https://gymnasium.farama.org

Open the following colab notebooks for examples:

Panda-Gym: https://colab.research.google.com/github/qgallouedec/panda-gym/blob/master/examples/PickAndPlace.ipynb

JAX: https://colab.research.google.com/github/RobertTLange/gymnax/blob/main/examples/00_getting_started.ipynb

DQN in Pytorch

Download Code Archive (unzip)