kenjyoung / dreamerv2_JAX

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

Related projects

Alternatives and complementary repositories for dreamerv2_JAX