Skip to main content
Ctrl+K

aimz

  • User Guide
  • Examples
  • FAQs
  • API Reference
    • Development
    • Changelog
  • GitHub
  • User Guide
  • Examples
  • FAQs
  • API Reference
  • Development
  • Changelog
  • GitHub

Section Navigation

  • KernelSpec
  • ImpactModel
    • aimz.ImpactModel
    • aimz.ImpactModel.inference
    • aimz.ImpactModel.kernel
    • aimz.ImpactModel.kernel_spec
    • aimz.ImpactModel.param_input
    • aimz.ImpactModel.param_output
    • aimz.ImpactModel.posterior
    • aimz.ImpactModel.rng_key
    • aimz.ImpactModel.temp_dir
    • aimz.ImpactModel.vi_result
    • aimz.ImpactModel.train_on_batch
    • aimz.ImpactModel.fit_on_batch
    • aimz.ImpactModel.fit
    • aimz.ImpactModel.set_posterior_sample
    • aimz.ImpactModel.is_fitted
    • aimz.ImpactModel.predict_on_batch
    • aimz.ImpactModel.predict
    • aimz.ImpactModel.log_likelihood
    • aimz.ImpactModel.sample_prior_predictive_on_batch
    • aimz.ImpactModel.sample_prior_predictive
    • aimz.ImpactModel.sample
    • aimz.ImpactModel.sample_posterior_predictive_on_batch
    • aimz.ImpactModel.sample_posterior_predictive
    • aimz.ImpactModel.estimate_effect
    • aimz.ImpactModel.cleanup
    • aimz.ImpactModel.cleanup_models
  • ArrayDataset
    • aimz.utils.data.ArrayDataset
  • ArrayLoader
    • aimz.utils.data.ArrayLoader
    • aimz.utils.data.ArrayLoader.pad_array
  • MLflow Integration
    • aimz.mlflow.autolog
    • aimz.mlflow.get_default_conda_env
    • aimz.mlflow.get_default_pip_requirements
    • aimz.mlflow.load_model
    • aimz.mlflow.log_model
    • aimz.mlflow.save_model
  • API Reference

API Reference#

  • KernelSpec
  • ImpactModel
  • ArrayDataset
  • ArrayLoader
  • MLflow Integration

previous

Frequently Asked Questions

next

KernelSpec
Edit on GitHub

© Copyright 2025-2026, Eli Lilly and Company.