Code release for Cole et al. (2021) "The functional relevance of task-state functional connectivity", Journal of Neuroscience