jax

d8c75da8 - update version and changelog for pypi