Neural Network Augmented Compartmental Pandemic Models