This library a simple interface to simulate mathematical epidemic models.
Simple SIR simulation
from epimodels.continuous.models import SIR
model = SIR()
model([1000, 1, 0], [0, 50], 1001, {'beta': 2, 'gamma': .1})
model.plot_traces()
For stochastic epidemic models check this.