Examples are very slow

[simsurace]

There appears to be a performance regression, leading to very slow performance in the examples. The examples CI step currently times out.

• exact time learning
• exact space-time learning
• approximate space-time learning

[simsurace]

I'll look into this soon

EDIT: @willtebbutt , I'll let you look into this first then