Skip to content

pump jax version to 0.1.21#21

Merged
neerajprad merged 1 commit intopyro-ppl:masterfrom
fehiepsi:version
Mar 6, 2019
Merged

pump jax version to 0.1.21#21
neerajprad merged 1 commit intopyro-ppl:masterfrom
fehiepsi:version

Conversation

@fehiepsi
Copy link
Member

@fehiepsi fehiepsi commented Mar 6, 2019

@neerajprad the lognorm test is failed with version 0.1.20 but passed with version 0.1.21 (I'm not sure which PR is changed during that). So I pump the requirement to the newer version. A nice feature of the new version is lax.cond. :)

@neerajprad
Copy link
Member

If we need to track jax more closely we can also directly import it as a submodule. We will need to modify our setup.py file to build jax from source in that case. Given the frequency with which they release the pip packages, we'll probably not need it though. :)

@neerajprad neerajprad merged commit ab25789 into pyro-ppl:master Mar 6, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants