Bump up version for release

README.md

@@ -59,7 +59,7 @@ Here are a few self-contained Colab notebooks to help you get started on using P
 Please consider citing our [companion paper](https://arxiv.org/abs/2202.04110) if you use PGMax in your work:
 ```
 @article{zhou2022pgmax,
-  author = {Zhou, Guangyao and Kumar, Nishanth and L{\'a}zaro-Gredilla, Miguel and Kushagra, Shrinu and George, Dileep},
+  author = {Zhou, Guangyao and Kumar, Nishanth and Dedieu, Antoine and L{\'a}zaro-Gredilla, Miguel and Kushagra, Shrinu and George, Dileep},
   title = {{PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX}},
   journal = {arXiv preprint arXiv:2202.04110},
   year={2022}

docs/source/conf.py

@@ -14,10 +14,10 @@
 
 project = "PGMax"
 copyright = "2021, Vicarious FPC Inc"
-author = "Guangyao Zhou, Nishanth Kumar, Miguel Lazaro-Gredilla, Shrinu Kushagra, Dileep George"
+author = "Guangyao Zhou, Nishanth Kumar, Antoine Dedieu, Miguel Lazaro-Gredilla, Shrinu Kushagra, Dileep George"
 
 # The full version, including alpha/beta/rc tags
-release = "0.3.0"
+release = "0.4.0"
 
 
 # -- General configuration ---------------------------------------------------

pgmax/__init__.py

@@ -1,3 +1,3 @@
 """A container package for the entire PGMax library."""
 
-__version__ = "0.3.0"
+__version__ = "0.4.0"