Skip to content

Handle renaming of pjit_p to jit_p#2052

Merged
fehiepsi merged 6 commits intopyro-ppl:masterfrom
tare:jit_p
Aug 4, 2025
Merged

Handle renaming of pjit_p to jit_p#2052
fehiepsi merged 6 commits intopyro-ppl:masterfrom
tare:jit_p

Conversation

@tare
Copy link
Copy Markdown
Contributor

@tare tare commented Jul 23, 2025

This PR aims to address #2051.

@tare
Copy link
Copy Markdown
Contributor Author

tare commented Jul 23, 2025

It looks like the CI errors are related to this.

@tare
Copy link
Copy Markdown
Contributor Author

tare commented Aug 3, 2025

Added change from #2056. It looks like we end up installing tensorflow_probability and tfp-nightly because jaxns requires tensorflow_probability (here), which still causes issues.

re: jaxns I created this issue and this PR.

Copy link
Copy Markdown
Member

@fehiepsi fehiepsi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry for the late response. Thank you so much for the fixes, @tare!

@fehiepsi fehiepsi merged commit ab77747 into pyro-ppl:master Aug 4, 2025
9 checks passed
@tare tare deleted the jit_p branch August 4, 2025 15:16
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