3 dépôts
GPU kernels that sample token indices from logits or probability distributions deterministically.
Distinct from Adaptive Probability Sampling: No candidate covers deterministic token sampling from logits; existing candidates focus on adaptive or distribution-based sampling.
Explore 3 awesome GitHub repositories matching artificial intelligence & ml · Deterministic Token Sampling Kernels. Refine with filters or upvote what's useful.
This project is a large language model inference library and framework designed to run models for text generation, problem solving, and coding assistance. It includes a multimodal framework for processing combined image and text inputs and a tool-use implementation that enables the execution of external functions based on model reasoning. The system features a distributed GPU inference engine that spreads large model workloads across multiple graphics processors to increase processing speed and meet memory requirements. It also provides containerized model deployment through pre-packaged imag
Selects tokens from a probability distribution using temperature and top-p filtering.
FlashInfer is a library of high-performance GPU kernels purpose-built for accelerating large language model inference. It provides optimized implementations for attention operations (including flash attention, page attention, multi-head latent attention, and cascade attention) using paged key-value caches, fused kernel composition, and just-in-time compilation. The library also includes specialized kernels for mixture-of-experts layers, block-scaled low-precision quantization (FP8, FP4), and distributed collective communication. What distinguishes FlashInfer is its fused all-reduce communicat
Provides deterministic and batched token sampling from logits on GPU.
YaLM-100B is a large language model and open-weights AI model designed for generating and processing natural language text. It functions as a multilingual text generator optimized for producing and understanding human language content specifically in English and Russian. The model is built for large scale language modeling and open source AI research, providing a foundation for text-based machine learning tasks. It utilizes a decoder-only transformer architecture with a multilingual embedding space to map English and Russian text into a shared vector space. Its broader capabilities cover nat
Employs probability distribution-based sampling and greedy selection to determine the final output tokens.