Project Awesome project awesome
♥

New Libraries > PGMax

A framework for building discrete Probabilistic Graphical Models (PGM's) and running inference inference on them via JAX.

Package 65 stars Archived GitHub

continuous-integration PyPI version pre-commit.ci status codecov Documentation Status

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.

Back to JAX