Jigsaw exploration

Created by: oskar-taubert

fixes positional embedding now converges on dummy data results for jigsaw on real data still abhorrent, but have not explored to deeply