Expand description
Adam optimizer step for the 2-D embedding.
All state operates on Vec<[f32; 2]> — no matrix types, no large allocations.
Rayon par_chunks_mut parallelises the per-point update.
Structs§
- Adam
State - Persistent Adam state buffers. Allocated once, live for the full run.
Functions§
- adam_
step - Apply one Adam step: update
embeddingin place.