Update __init__.py

Author: thomaspinder

gpjax/__init__.py

@@ -42,7 +42,7 @@
 __description__ = "Didactic Gaussian processes in JAX"
 __url__ = "https://github.com/thomaspinder/GPJax"
 __contributors__ = "https://github.com/thomaspinder/GPJax/graphs/contributors"
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 
 
 __all__ = [