Just uploaded codes implementing Economy 4 of Borella, De Nardi & Yang (2018) - The aggregate implications of gender and marriage.
Codes: https://github.com/robertdkirkby/LifeCycleOLGReadingList/tree/main/BorellaDeNardiYang2018
One step involves putting agents onto the initial distribution as a (log) trivariate-normal distribution. There is a command in VFI Toolkit for putting multinomial distributions on a grid (i.e. grid is input, probabilities on the grid are output) to do this which has not previously been documented anywhere.
[Original paper was missing some parameters, but authors were great about sending them by email. You can find them all in the code.]
[Requires a fairly powerful gpu to run as is, but if you just want to play with it then make the grids smaller and will be fine on standard desktop gpu.]