/dreamerv2_JAX

An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.

Primary LanguagePythonMIT LicenseMIT

Watchers