With many different plots the plotting of all the matplotlib components takes up significant time in the visualisations. An easy and simple way to speed this up would be only plotting the graphs every N steps. We could add a slider to the visualisation that defaults to 1 and allows updating the graphs less frequently for performance.