Any appetite for a quadratic approximation PR?

I think it should be :slight_smile:

I made an example you can check out here https://colab.research.google.com/drive/1DTe7QchyW-wpbUmulzY27lBTRFQXzhm0?usp=sharing