// ============================================================================
// Core Loader
// ============================================================================

export {
  DopplerLoader,
  getDopplerLoader,
  createDopplerLoader,
} from './doppler-loader.js';

export type {
  TensorLocation,
  LayerWeights,
  LoadProgress,
  LoadOptions,
  CustomShardLoader,
  CustomShardLoaderOptions,
  LoaderStats,
  KernelCapabilities,
  Q4KConfig,
  ModelConfig,
  ShardLoadPriority,
  ShardLoadOptions,
  CustomShardRangeLoader,
  CustomShardStreamLoader,
  CustomShardStreamOptions,
  ShardSourceInfo,
} from './loader-types.js';

export { MultiModelLoader } from './multi-model-loader.js';
export type { AdapterSource } from './multi-model-loader.js';

export { LoaderState, createLoaderState } from './loader-state.js';
export type { WeightType, LoaderStateSnapshot } from './loader-state.js';

// ============================================================================
// Manifest & Shard Resolution
// ============================================================================

export {
  needsNormWeightOffset,
  getLargeWeightConfig,
  getLargeWeightMaxBytes,
  estimateMatmulWeightBytes,
  resolveMatmulWeightDtype,
  requiresCpuF16ToF32MatmulMaterialization,
  resolveWeightLayout,
  shouldStreamLargeWeight,
  isMoEModel,
} from './manifest-config.js';
export type { LargeWeightConfigSchema as LargeWeightConfig } from '../config/schema/doppler.schema.js';

export { buildTensorLocations } from './shard-resolver.js';
export type { BuildTensorLocationsOptions } from './shard-resolver.js';

export { ShardCache, createShardCache } from './shard-cache.js';
export type { ShardCacheConfig } from './shard-cache.js';

// ============================================================================
// Expert Loading & Caching
// ============================================================================

export {
  preloadShardsForExpert,
  prefetchExperts,
  predictNextLayerExperts,
  loadExpert,
} from './experts/expert-loader.js';

export {
  ExpertCache,
  getExpertCache,
  createExpertCache,
} from './experts/expert-cache.js';
export type { CacheStats } from './experts/expert-cache.js';

// ============================================================================
// Tensor Loading
// ============================================================================

export {
  isPackedQ4K,
  shouldUseFusedQ4K,
  getQ4KOutputDtype,
  getWeightLayout,
  convertBF16ToF32CPU,
  convertF16ToF32CPU,
  loadQ4KFused,
  loadQ4KDequant,
  loadQ6K,
  loadBF16,
  loadFloat,
  loadTensorToGPU,
  loadTensorToCPU,
} from './tensors/tensor-loader.js';
export type { TensorLoadConfig, TensorLoadResult } from './tensors/tensor-loader.js';

export { assembleShardData } from './tensors/tensor-reader.js';

export { getTensorNamesByRole } from './tensors/tensor-role.js';

// ============================================================================
// Weight Stage Loaders
// ============================================================================

export { loadEmbeddings } from './embedding-loader.js';

export { loadLayer } from './layer-loader.js';

export { loadFinalWeights } from './final-weights-loader.js';

// ============================================================================
// Utilities
// ============================================================================

export {
  f16ToF32,
  convertBF16ToF32GPU,
  shouldDequantizeToF16,
  applyBufferLayout,
} from './dtype-utils.js';

export {
  maybeDowncastToF16,
  batchDowncastWeights,
} from './weight-downcast.js';

export {
  captureMemorySnapshot,
  formatMemoryStats,
  MemoryMonitor,
  MemoryTimeSeries,
} from './memory-monitor.js';
export type { MemorySnapshot } from './memory-monitor.js';

export {
  QK_K,
  QK4_K_BLOCK_SIZE,
  K_SCALE_SIZE,
  Q4K_BLOCK_BYTES,
  Q6K_BLOCK_BYTES,
  Q8_0_BLOCK_BYTES,
  Q8_0_BLOCK_SIZE,
  padToQ4KBlock,
  q4kBlockCount,
} from './quantization-constants.js';

// ============================================================================
// Type-only re-exports
// ============================================================================

export type { ExpertWeights } from './weights.js';