hiermodelutils

Utilities for probabilistic hierarchical modeling.

Overview

In a nutshell, this package is a hodgepodge collection of useful utilities for...

• ...storing/retrieving hyperparameters for complex, hierarchical, probabilistic models (i.e., using Numpyro and Equinox).
• ...formatting hierarchical data to integrate smoothly with Numpyro.
• ...doing vectorized computations with missing data efficiently.
• ...and other random stuff (custom defined Numpyro distributions, probability density function visualization, etc.).

While the focus is on supporting compatibility with Numpyro's API, most of the utilities are actually agnostic to the underlying probabilistic programming framework.

Examples

• TODO: Add examples

Installation

pip install git+https://github.com/wesleyjholt/hiermodelutils.git

Usage

import hiermodelutils as hmu

TODO: Add example