Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update host_callback used in blackjax #562

Open
junpenglao opened this issue Aug 8, 2023 · 3 comments
Open

Update host_callback used in blackjax #562

junpenglao opened this issue Aug 8, 2023 · 3 comments
Labels
good first issue Good for newcomers

Comments

@junpenglao
Copy link
Member

junpenglao commented Aug 8, 2023 •
edited

For background see google/jax#13555

jax.experimental.host_callback is deprecated, and we should switch to using jax.debug.print/jax.debug.callback in blackjax/progress_bar.py
@junpenglao junpenglao added the good first issue Good for newcomers label Aug 8, 2023
@junpenglao
Copy link
Member Author

Will need to migrate to eitherdebug.callback, io_callback, or pure_callback

@junpenglao
Copy link
Member Author

Closed in #651

@junpenglao junpenglao changed the title Update host_callback used in blackjax/progress_bar.py Update host_callback used in blackjax Mar 28, 2024
@junpenglao
Copy link
Member Author

Leaving this open as we need to also update https://blackjax-devs.github.io/blackjax/examples/howto_other_frameworks.html

@junpenglao junpenglao reopened this Mar 28, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
good first issue Good for newcomers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@junpenglao