jaxili.inference package

Inference.

This module contains the inference classes for the JaxILI library. JaxILI implements classes for Neural Posterior and Likelihood Estimations.

Submodules

• jaxili.inference.nle module
  • NLE
    • NLE.append_simulations()
    • NLE.build_posterior()
    • NLE.create_trainer()
    • NLE.load_from_checkpoints()
    • NLE.set_dataloader()
    • NLE.set_dataset()
    • NLE.set_loss_fn()
    • NLE.set_model_hparams()
    • NLE.train()
• jaxili.inference.npe module
  • NPE
    • NPE.append_simulations()
    • NPE.build_posterior()
    • NPE.create_trainer()
    • NPE.load_from_checkpoints()
    • NPE.set_dataloader()
    • NPE.set_dataset()
    • NPE.set_loss_fn()
    • NPE.set_model_hparams()
    • NPE.train()