IQL-PyTorch

This repository houses a minimal PyTorch implementation of Implicit Q-Learning (IQL), an offline reinforcement learning algorithm, along with a script to run IQL on tasks from the D4RL benchmark.

Note that the paper’s authors have published their official implementation, which is based on JAX. My implementation is intended to be an alternative for PyTorch users, and my general recommendation is to use the authors’ code unless you specifically want/need PyTorch for some reason. I am planning to validate my implementation against the results stated in the paper once I have some spare compute.

GitHub

View Github