search

jeremiecoullon / SGMCMCJax

Lightweight library of stochastic gradient MCMC algorithms written in JAX.
https://sgmcmcjax.readthedocs.io/en/latest/index.html
Apache License 2.0
95 stars 8 forks source link

Comparison to other packages in JOSS #31

Closed canyon289 closed 2 years ago

canyon289 commented 2 years ago

State of the field: Do the authors describe how this software compares to other commonly-used packages?

In the paper I believe it may be worthwhile to explain how this package differs from other package such as numpyro, stan, PyMC etc. This could detail the difference in sampling algorithms, API, and extensibility.

I also am curious about this line. What purpose would an integration serve, since integration is mentioned.

can integrate easily with other codebases within the JAX ecosystem such as Flax, a neural network library for JAX

jeremiecoullon commented 2 years ago

I made these changes to paper; let me know if you think I should clarify this some more!

canyon289 commented 2 years ago

Looks great as well. You state it succinctly and well. Thank you