I am working on a mixture model with family-level random effects. I would like to parameterize the random effects as independent across families each with a covariance matrix parameterized using the kernel trick (Gaussian Process Regression). Each family has a distance matrix \mathbf{D}_j (of potentially differing dimension) that is converted into the covariance matrix with a Gaussian Kernel: \mathbf{h}_j\sim\mathcal{N}\left(\mathbf{0}, \tau\exp\left[-\rho\mathbf{D}_j\right]\right), where the exp() is applied element-wise. I know that ragged arrays have yet to be implemented (https://github.com/stan-dev/stan/wiki/Ragged-array-spec) but in R, we could simply create a list called D where D[[j]] accesses \mathbf{D}_j. Does anyone have ideas about how to (simply) implement something like this in RStan without using very large, sparse matrices? All I need is to define \mathbf{K}_j=\exp\left[-\rho\mathbf{D}_j\right] in the transformed parameters block for learning \rho, which requires a way of easily constructing/accessing each \mathbf{D}_j.

How to Use Ragged Arrays for Matrices?

rsmall March 22, 2019, 3:31am 2

For a solution to this question see bgoodri’s answer on Is Ragged Array allowed in Stan?

Topic		Replies	Views
Is Ragged Array allowed in Stan? General	20	3223	March 13, 2023
(possibly bad) Idea for (debatably) easier ragged arrays General specification	10	871	October 17, 2020
Sparse GPs of different sizes Modeling	3	499	February 13, 2023
Ragged array expressions Modeling	17	2091	October 22, 2018
Problem when converting to a ragged array model (unsure why - syntax? typo?) Modeling specification	6	1133	June 14, 2017