In a more compact form we have
which is the Fokker-Planck equation! It is trivial to replace position with velocity (momentum).