Important note: PGMax is now part of DeepMind 🎉!

Following the acquisition of Vicarious by Alphabet in 2022, we are excited to share that PGMax has moved to DeepMind! This repository is no longer being maintained, and all future developments will happen in the new PGMax repository under DeepMind.

We look forward to the new exciting research possibilities from PGMax's integration within DeepMind's rapidly growing JAX ecosystem.