Refactor ATOM for top-k top-p sampling support#227

Open

aryaman-gupta wants to merge 6 commits intomainfrom

aryaman/topk-topp-support

Commits on Feb 20, 2026

refactor for top-k top-p sampling support
aryaman-gupta
committed

Commits on Feb 23, 2026

sampler.py: change in comments and default param syntax
aryaman-gupta
committed

Commits on Feb 24, 2026

sampler.py: adds fast path for all temperature=0
aryaman-gupta
committed

Commits on Feb 25, 2026

sampler.py: adds warning for native Pytorch implementation of top-k top-p
aryaman-gupta
committed

Commits on Feb 26, 2026