A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work - View it on GitHub