Measurement of photon-jet transverse momentum correlations in 5.02 TeV Pb + Pb and pp collisions with ATLAS