Code used in the analyses for
Hierarchical recurrent state space models reveal discrete and continuous dynamics of neural activity in C. elegans
Scott Linderman, Annika Nichols, David Blei, Manuel Zimmer, Liam Paninski
bioRxiv 621540; doi: https://doi.org/10.1101/621540
Note: This code is very rough around the edges and may no longer work due to changes in SSM.
Someday I hope to revisit these models using dynamax.