JAX_VISR JAX implementation of VISR (ICLR - Hansen et al. 2020) Usage e.g. python VISR.py --game Breakout --recurrent 1 Requirements flax gymnax jax matplotlib numpy optax exact versions used can be found in requirements.txt Baselines https://wandb.ai/mjsargent/tSF_VISR/reports/VISR-Baselines--VmlldzoyNzMzMTU0?accessToken=n7zrkqw7zbj7tm7fnuvlf7czkeszwjb7cu0ctkl2kv7ywav67e7qlad6c2v3pki0