Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.Samplers.TokenSampler Module Reference

Classes

class  Mila::Dnn::TokenSampler< TDeviceType, TPrecision >
 The standard token sampler: temperature / top-k / top-p multinomial. More...

Files

file  Mila/Src/Dnn/Samplers/TokenSampler.ixx
 Device-agnostic token sampler facade.