The chemical element that imparts the yellow color observed in firework displays is sodium (Na).

When sodium salts are heated in a flame, the electrons in the sodium atoms become excited and jump to higher energy levels. As they return to their ground state, they emit light at a specific wavelength that corresponds to the yellow color we see.