Suppose that you have many countries, indexed by i, who are identical in all margins except they have different levels of productivity which are assumed constant across time but which differ across countries. We denote these levels of productivity by Ai. The central equation governing the dynamics of capital per worker in a country i is given by