Learning Quantum Graphical Models using Constrained Gradient Descent on the Stiefel Manifold