Is your feature request related to a problem? Please describe.

We want to do masked token modelling pre-training with interpolated latent noise following https://arxiv.org/abs/2507.15856

Describe the solution you'd like

No response

Describe alternatives you've considered

No response

Additional context

No response

Organisation

No response