Исходя из специальной теории относительности Эйнштейн в том же 1905 году открыл форму взаимосвязи массы и энергии. Из нее следует, что любой перенос энергии связан с переносом массы. Эта формула трактуется так же, как выражение, описывающее «превращение» массы в энергию. Именно на этом представлении основано объяснение так называемого «дефекта массы». В конце 1909 года Эйнштейн получил место экстраординарного профессора теоретической физики Цюрихского университета. Здесь он преподавал только три семестра, затем последовало почетное приглашение на кафедру теоретической физики Немецкого университета в Праге.
Исходя из своего принципа относительности он в 1911 году в статье «О влиянии силы тяжести на распространение света» заложил основы релятивистской теории тяготения, высказав мысль, что световые лучи, испускаемые звездами и проходящие вблизи Солнца, должны изгибаться у его поверхности. Таким образом, предполагалось, что свет обладает инерцией и в поле тяготения Солнца должен испытывать сильное гравитационное воздействие. Летом 1912 года Эйнштейн вернулся в Цюрих, где в Высшей технической школе была создана кафедра математической физики. Здесь он занялся разработкой математического аппарата, необходимого для дальнейшего развития теории относительности. В этом ему помогал его ученик Марсель Гросман. Плодом их совместных усилий стал труд «Проект обобщенной теории относительности и теории тяготения». В Берлин Эйнштейн прибыл в апреле 1914 года, будучи уже членом Академии наук (1913), и приступил к работе в созданном Гумбольдтом университете — крупнейшем высшем учебном заведении Германии. Здесь он провел 19 лет — читал лекции, вел семинары, регулярно участвовал в работе коллоквиума, который во время учебного года раз в неделю проводился в Физическом институте.
В 1915 году Эйнштейн завершил создание общей теории относительности. Всего через год после опубликования работы по общей теории относительности Эйнштейн представил еще одну работу, имеющую революционное значение. Поскольку не существует пространства и времени без материи, то есть без вещества и поля, отсюда с необходимостью следует, что Вселенная должна быть пространственно конечной. В 1916–1917 годах вышли работы Эйнштейна, посвященные квантовой теории излучения. В них он рассмотрел вероятности переходов между стационарными состояниями атома (теория Н. Бора) и выдвинул идею индуцированного излучения. Эта концепция стала теоретической основой современной лазерной техники.
Середина 1920-х годов ознаменовалась в физике созданием квантовой механики. Несмотря на то что идеи Эйнштейна во многом способствовали ее становлению, вскоре обнаружились значительные расхождения между ним и ведущими представителями квантовой механики. Эйнштейн не мог примириться с тем, что закономерности микромира носят лишь вероятностный характер. Между тем политическая ситуация в Германии становилась все более напряженной. Вскоре началась планомерная кампания против создателя теории относительности. В начале 1933 года Эйнштейн находился в Пасадене, и после прихода Гитлера к власти он никогда более не ступал на немецкую землю. В марте 1933 года заявил о своем выходе из Прусской академии наук и отказался от прусского гражданства.
С октября 1933 года Эйнштейн приступил к работе в Принстонском университете, а вскоре получил американское гражданство, одновременно оставаясь гражданином Швейцарии. Ученый продолжал свои работы по теории относительности; большое внимание уделял попыткам создания единой теории поля.
Находясь в США, ученый старался любыми доступными ему средствами оказывать моральную и материальную поддержку немецким антифашистам. Его очень беспокоило развитие политической ситуации в Германии. Эйнштейн опасался, что после открытия деления ядра Ганом и Штрассманом у Гитлера появится атомное оружие. Тревожась за судьбу мира, Эйнштейн направил президенту США Ф. Рузвельту свое знаменитое письмо, которое побудило последнего приступить к работам по созданию атомного оружия.
После окончания второй мировой войны Эйнштейн включился в борьбу за всеобщее разоружение. На торжественном заседании сессии ООН в Нью-Йорке в 1947 году он заявил об ответственности ученых за судьбы мира, а в 1948 году выступил с обращением, в котором призывал к запрещению оружия массового поражения. Мирное сосуществование, запрещение ядерного оружия, борьба против пропаганды войны — эти вопросы занимали Эйнштейна в последние годы его жизни не меньше, чем физика.
Умер Эйнштейн в Принстоне (США) 18 апреля 1955 года. Его прах был развеян друзьями в месте, которое должно навсегда остаться неизвестным.