Etter vel ett år i Praha vendte Einstein tilbake til Sveits da han ble tilbudt et professorat ved ETH i Zürich. Der inngikk han et samarbeid med matematikeren Marcel Grossmann for å forstå hvordan Riemanns differensialgeometri kunne benyttes til å beskrive gravitasjonskrefter. Einstein så for seg hvordan en masse, for eksempel en planet eller Solen, ville krumme rommet omkring seg slik at lys og partikler ville få en annen bevegelse enn i Newtonsk fysikk.