Regresó a Shanghai en 1946 para fundar el Instituto de Matemáticas de la Academia Sínica, que más tarde fue trasladado a Nanking.