Custom topk-logprobs kernel + Remove a redundant Device to Host logits copy.#434

davide221 merged 4 commits into

Luce-Org:mainLuce-Org/lucebox-hub:mainfrom

GeometricAGI:geometric-ai/optimize_topkGeometricAGI/lucebox-hub:geometric-ai/optimize_topkCopy head branch name to clipboard