Reference

• jaxili.inference package
  • Submodules
• jaxili.loss module
  • gaussian_kernel_matrix()
  • loss_mmd_npe()
  • loss_nll_nle()
  • loss_nll_npe()
  • maximum_mean_discrepancy()
  • mmd_kernel()
  • mmd_summary_space()
• jaxili.model module
  • AffineCoupling
  • Compressor_w_NDE
  • ConditionalMADE
  • ConditionalMAF
  • ConditionalRealNVP
  • MAFLayer
  • MaskedLinear
  • MixtureDensityNetwork
  • NDENetwork
  • NDE_Compressor
  • NDE_w_Standardization
• jaxili.posterior package
  • Submodules
• jaxili.train module
  • TrainState
  • TrainerModule
• jaxili.utils module
  • check_density_estimator()
  • check_hparams_maf()
  • check_hparams_mdn()
  • check_hparams_realnvp()
  • create_data_loader()
  • handle_non_serializable()
  • validate_theta_x()
• jaxili.validation module