Sampler
Add noise
×no parameters
Add fresh Gaussian noise scaled by the schedule’s rate and the length of the step.
- in
- [B batch, T positions, 768 d] [B batch]
- out
- [B batch, T positions, 768 d]
Explanation
This is the Euler–Maruyama step: the mean follows the drift and the noise follows the diffusion coefficient .
Sources2
Direct evidence
One Euler–Maruyama step: the mean is x + drift · dt, and the next state adds diffusion · √(−dt) · noise; the clean estimate is returned with them.
The implementation, pinned at revision e18c42acdfb7, not yet public
Direct evidence
The reverse SDE: drift −β x/2 − β · score and diffusion √β; the ODE halves the score term and has no diffusion.
The implementation, pinned at revision e18c42acdfb7, not yet public