Log metrics for tensorboard

rlouf / mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.

https://rlouf.github.io/mcx

Apache License 2.0

325 stars 17 forks source link

Log metrics for tensorboard #34

Open rlouf opened 3 years ago

rlouf commented 3 years ago

It would be nice to be able to monitor the progress of sampling in real time. Totally do-able either using a callback and/or by logging every X steps when using "generate".

ahartikainen commented 3 years ago

Hi, I think you could be interested in developing this https://github.com/arviz-devs/arviz_dashboard

See the wiki.

rlouf commented 3 years ago

This looks very interesting, I will have a look!