Variational Flow Matching realization from the paper "Variational Flow Matching for Graph Generation"
