A prototype of polio toy model:

• Simple S, I, R with constant population by village
• Connecting villages in explicit neighbor network
• Tracking individual infections -- simple fixed generation time + infectiousness
• Reporting summary statistics
• Reporting detailed transmission events

And some interactive plotly visualization to explore infection and immune dynamics:

And a randomly subsampled tree visualized with tskit