Sampling API Reference

API reference for sampling methods and utilities in Artifex.

Overview

This module provides sampling utilities for generating outputs from trained generative models.

Core Sampling Functions

Basic Sampling

from artifex.generative_models.core.sampling import sample_from_model

samples = sample_from_model(
    model=model,
    num_samples=64,
    rng=jax.random.key(0),
)

Temperature Scaling

from artifex.generative_models.core.sampling import temperature_sample

samples = temperature_sample(
    model=model,
    num_samples=64,
    temperature=0.8,
    rng=jax.random.key(0),
)

Model-Specific Sampling

For detailed sampling methods for each model type, see:

• VAE Sampling
• GAN Sampling
• Diffusion Sampling
• Flow Sampling

Related Documentation

• Sampling User Guide - Comprehensive sampling guide
• Inference Optimization - Performance optimization
• Core API - Core module reference