index.ts

/**
 * TypeScript Sampling Library
 *
 * Pure TS implementations of sampling algorithms.
 * Designed for OTA updates, custom strategies, and transparency.
 *
 * Performance: ~0.5-1.0ms per token (vs ~0.1-0.3ms native)
 * Trade-off: <2% overhead for full flexibility + debuggability
 */

// Core sampling strategies
export { greedy } from './greedy.js';
export { softmax } from './softmax.js';

// Combined strategies
export { sampleWithStrategy, getTopCandidates, type SamplerMode, type SamplingParams, type SampleOptions } from './strategies.js';

// Functional filters (for advanced use / testing)
export {
  applyTopK,
  applyTypicalP,
  applyTopP,
  applyMinP,
  applyTopNSigma,
  applyTemperature,
  sampleFromSet,
  candidateSetFromFullLogits,
  CandidateSet,
} from './filters.js';

// Penalties
export {
  TokenHistoryTracker,
  applyRepetitionPenalty,
  applyFrequencyPenalty,
  applyPresencePenalty,
  applyPenalties,
} from './penalties.js';

// PRNG (deterministic random number generation)
export {
  Xoroshiro128Plus,
  initializePRNG,
  random,
  resetPRNG,
} from './prng.js';

// Native fast path (opt-in)

// Utilities
export { SamplerWorkspace, selectTopK, requiredKcap, DEFAULT_KCAP, TYPICAL_P_KCAP } from './utils.js';

// Metrics (for Test-Time Alignment)
export {
  computeModelSurprisal,
  computeSamplingSurprisal,
  computeModelEntropy,
  computeSamplingEntropy,
  RollingPerplexity,
  type SurprisalBase,
} from './metrics.js';