JAX_VISR

JAX implementation of VISR (ICLR - Hansen et al. 2020)

Usage

e.g. python VISR.py --game Breakout --recurrent 1

flax gymnax jax matplotlib numpy optax exact versions used can be found in requirements.txt

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
LICENSE		LICENSE
README.md		README.md
VISR.py		VISR.py
networks.py		networks.py
requirements.txt		requirements.txt
utils.py		utils.py