More generally, given that you are emphasizing the lifting aspect, non-linear lifting was shown to be crucial for universal approx of FNO even if you kept one mode in lifted space:

→ View original post on X — @animaanandkumar