Designing and implementing new interface for Factor Graph specificati…

…on (#30) * deletes contrib module * updates README to include cool new PyPI version badge :) * includes new CompositeVariable dataclass * gets rid of new_interface_tests and integrates code into sanity_check_example * Update pgmax/interface/datatypes.py Co-authored-by: Guangyao Zhou <[email protected]> * small updates to datatypes * adds more docstrings and resolves minor errors in datatypes Co-authored-by: Guangyao Zhou <[email protected]>
vicariousinc · Aug 2, 2021 · e1f130f · e1f130f
1 parent 40efc27
commit e1f130f
Show file tree

Hide file tree

Showing 21 changed files with 487 additions and 6,990 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,5 @@
 [![continuous-integration](https://github.com/vicariousinc/PGMax/actions/workflows/ci.yaml/badge.svg)](https://github.com/vicariousinc/PGMax/actions/workflows/ci.yaml)
+[![PyPI version](https://badge.fury.io/py/pgmax.svg)](https://badge.fury.io/py/pgmax)
 
 # PGMax
 PGMax is a library for working with Factor Graphs in [JAX](https://jax.readthedocs.io/en/latest/). It currently provides an interface for specifying factor graphs of any type, as well as an efficient implementation of max-product belief propagation and inference on these graphs.