If gravity can be thought of as both a wave (the gravitational wave, as predicted to exist by Albert Einstein and certain calculations) and a particle (the graviton), would it make sense to apply quantum mechanics (which I understand only applies to mass/energy) and therefore wavefunction collapse to gravity? In other words, does gravity exhibit wave-particle duality as light does, and thus is it susceptible to wavefunction collapse? If so, what would the implications of the wavefunction collapse of a gravitational wave be?
To better sum up my question: could a gravitational wave be described as a wavefunction?
I would appreciate it if anyone could help me understand if this is a valid concept, or if there are any other theories and concepts that would help me understand gravity and quantum mechanics combined (quantum field theory?).