How can I use qaffeine the same way as eqcss, with @element?

[deleite]

Hi, I used EQCSS and you can see how it look on the sample below.
https://codepen.io/deleite/pen/dyMryGY

Issue is that with EQCSS there is a flickering before the layout switch, very briefly but happens.

I am trying to use the same with qaffeine, and it works by applying the styles from the js without the flickering.
https://codepen.io/deleite/pen/MWyxWmZ

but how can I use it with the @element in the css?