Open ~/.bash_profile in your favorite editor and add the following content to the bottom.
# Git branch in prompt.
parse_git_branch() {| from jax import random, jit | |
| import jax.numpy as jnp | |
| from jax.scipy import stats | |
| from util import ravelize_function, make_log_density | |
| __all__ = ["log_density", "log_density_vec", "init_draw_zero"] | |
| def constrain_parameters(sigma_unc, alpha, beta): |