Pinned Loading

1. compmem/cognax compmem/cognax Public

   Build, fit, and sample from cognitive models with JAX + NumPyro.

   Python 1

2. Functional NumPyro MCMC Wrapper Functional NumPyro MCMC Wrapper

   1

   from functools import partial

   2

   from operator import attrgetter

   3

   4

   import jax

   5

   import jax.numpy as jnp

3. JAXCat JAXCat Public

   An implementation of CrossCat fit with MFVI.

   Python

4. CrossCatGibbs CrossCatGibbs Public

   My homespun implementation of CrossCat, fit with MCMC.

   Jupyter Notebook

5. LatentSimplexPosition LatentSimplexPosition Public

   An illustration of the Latent Simplex Position model.

   Jupyter Notebook