This is an influential paper in the macro-public finance literature:
Sebastian Dyrda, Marcelo Pedroni, Optimal Fiscal Policy in a Model with Uninsurable Idiosyncratic Income Risk, The Review of Economic Studies , Volume 90, Issue 2, March 2023, Pages 744–780, https://doi.org/10.1093/restud/rdac031
The model is relatively simple: Aiyagari with endogenous labor supply, proportional capital and labor income taxes, government debt (as in Aiyagari and McGrattan 1998). The interesting part, from a computational (and economic) point of view, is the transition.
Replicating this paper could be a great exercise to test the toolkit ability in dealing with transition paths.