Attempt to replicate Inductive Moment Matching (https://arxiv.org/abs/2503.07565) - View it on GitHub