jaxtronomy package

Subpackages

• jaxtronomy.Data package
  • Submodules
  • jaxtronomy.Data.image_noise module
    • ImageNoise
      • ImageNoise.__init__()
      • ImageNoise.C_D_model()
    • covariance_matrix()
  • jaxtronomy.Data.imaging_data module
    • ImageData
      • ImageData.__init__()
      • ImageData.update_data()
      • ImageData.likelihood_method()
  • Module contents
• jaxtronomy.ImSim package
  • Subpackages
    • jaxtronomy.ImSim.MultiBand package
      • Submodules
      • jaxtronomy.ImSim.MultiBand.single_band_multi_model module
      • Module contents
    • jaxtronomy.ImSim.Numerics package
      • Submodules
      • jaxtronomy.ImSim.Numerics.convolution module
      • jaxtronomy.ImSim.Numerics.grid module
      • jaxtronomy.ImSim.Numerics.numerics module
      • jaxtronomy.ImSim.Numerics.numerics_subframe module
      • jaxtronomy.ImSim.Numerics.point_source_rendering module
      • Module contents
  • Submodules
  • jaxtronomy.ImSim.de_lens module
    • get_param_WLS()
    • marginalisation_const()
    • marginalization_new()
  • jaxtronomy.ImSim.image_linear_solve module
    • ImageLinearFit
      • ImageLinearFit.__init__()
      • ImageLinearFit.image_linear_solve()
      • ImageLinearFit.likelihood_data_given_model()
      • ImageLinearFit.likelihood_data_given_model_solution()
      • ImageLinearFit.num_param_linear()
      • ImageLinearFit.linear_response_matrix()
      • ImageLinearFit.update_linear_kwargs()
      • ImageLinearFit.linear_param_from_kwargs()
      • ImageLinearFit.error_map_source()
      • ImageLinearFit.point_source_linear_response_set()
      • ImageLinearFit.check_positive_flux()
  • jaxtronomy.ImSim.image_model module
    • ImageModel
      • ImageModel.__init__()
      • ImageModel.likelihood_data_given_model()
      • ImageModel.source_surface_brightness()
      • ImageModel.lens_surface_brightness()
      • ImageModel.point_source()
      • ImageModel.image()
      • ImageModel.reduced_residuals()
      • ImageModel.reduced_chi2()
      • ImageModel.image2array_masked()
      • ImageModel.array_masked2image()
      • ImageModel.data_response
      • ImageModel.error_response()
      • ImageModel.update_psf()
      • ImageModel.update_data()
  • Module contents
• jaxtronomy.LensModel package
  • Subpackages
    • jaxtronomy.LensModel.LineOfSight package
      • Subpackages
      • Submodules
      • jaxtronomy.LensModel.LineOfSight.single_plane_los module
      • Module contents
    • jaxtronomy.LensModel.MultiPlane package
      • Submodules
      • jaxtronomy.LensModel.MultiPlane.decoupled_multi_plane module
      • Module contents
    • jaxtronomy.LensModel.Profiles package
      • Submodules
      • jaxtronomy.LensModel.Profiles.convergence module
      • jaxtronomy.LensModel.Profiles.cored_steep_ellipsoid module
      • jaxtronomy.LensModel.Profiles.epl module
      • jaxtronomy.LensModel.Profiles.epl_multipole_m1m3m4 module
      • jaxtronomy.LensModel.Profiles.epl_multipole_m3m4 module
      • jaxtronomy.LensModel.Profiles.gaussian module
      • jaxtronomy.LensModel.Profiles.gaussian_potential module
      • jaxtronomy.LensModel.Profiles.hernquist module
      • jaxtronomy.LensModel.Profiles.hernquist_ellipse_cse module
      • jaxtronomy.LensModel.Profiles.multipole module
      • jaxtronomy.LensModel.Profiles.nfw module
      • jaxtronomy.LensModel.Profiles.nfw_ellipse_cse module
      • jaxtronomy.LensModel.Profiles.nie module
      • jaxtronomy.LensModel.Profiles.pseudo_jaffe module
      • jaxtronomy.LensModel.Profiles.pseudo_jaffe_ellipse_potential module
      • jaxtronomy.LensModel.Profiles.sersic_utils module
      • jaxtronomy.LensModel.Profiles.shear module
      • jaxtronomy.LensModel.Profiles.sie module
      • jaxtronomy.LensModel.Profiles.sis module
      • jaxtronomy.LensModel.Profiles.spp module
      • jaxtronomy.LensModel.Profiles.tnfw module
      • Module contents
    • jaxtronomy.LensModel.Solver package
      • Submodules
      • jaxtronomy.LensModel.Solver.lens_equation_solver module
      • Module contents
    • jaxtronomy.LensModel.Util package
      • Submodules
      • jaxtronomy.LensModel.Util.decouple_multi_plane_util module
      • Module contents
  • Submodules
  • jaxtronomy.LensModel.lens_model module
    • LensModel
      • LensModel.__init__()
      • LensModel.info()
      • LensModel.check_parameters()
      • LensModel.ray_shooting()
      • LensModel.fermat_potential()
      • LensModel.arrival_time()
      • LensModel.potential()
      • LensModel.alpha()
      • LensModel.hessian()
      • LensModel.kappa()
      • LensModel.curl()
      • LensModel.gamma()
      • LensModel.magnification()
      • LensModel.flexion()
      • LensModel.set_static()
      • LensModel.set_dynamic()
      • LensModel.ddt_scaling
      • LensModel.update_cosmology()
  • jaxtronomy.LensModel.profile_list_base module
    • ProfileListBase
      • ProfileListBase.__init__()
      • ProfileListBase.set_static()
      • ProfileListBase.set_dynamic()
      • ProfileListBase.model_info()
      • ProfileListBase.check_parameters()
  • jaxtronomy.LensModel.single_plane module
    • SinglePlane
      • SinglePlane.__init__()
      • SinglePlane.ray_shooting()
      • SinglePlane.fermat_potential()
      • SinglePlane.potential()
      • SinglePlane.alpha()
      • SinglePlane.hessian()
      • SinglePlane.change_redshift_scaling()
      • SinglePlane.alpha_scaling
      • SinglePlane.mass_3d()
      • SinglePlane.mass_2d()
      • SinglePlane.density()
  • Module contents
• jaxtronomy.LightModel package
  • Subpackages
    • jaxtronomy.LightModel.Profiles package
      • Submodules
      • jaxtronomy.LightModel.Profiles.core_sersic module
      • jaxtronomy.LightModel.Profiles.gaussian module
      • jaxtronomy.LightModel.Profiles.sersic module
      • jaxtronomy.LightModel.Profiles.sersic_ellipse module
      • jaxtronomy.LightModel.Profiles.shapelets module
      • Module contents
  • Submodules
  • jaxtronomy.LightModel.light_model module
    • LightModel
      • LightModel.__init__()
  • jaxtronomy.LightModel.light_model_base module
    • LightModelBase
      • LightModelBase.__init__()
      • LightModelBase.surface_brightness()
      • LightModelBase.total_flux()
      • LightModelBase.param_name_list
      • LightModelBase.param_name_list_latex
      • LightModelBase.check_parameters()
  • jaxtronomy.LightModel.linear_basis module
    • LinearBasis
      • LinearBasis.__init__()
      • LinearBasis.functions_split()
      • LinearBasis.num_param_linear()
      • LinearBasis.num_param_linear_list()
      • LinearBasis.update_linear()
      • LinearBasis.add_fixed_linear()
      • LinearBasis.linear_param_from_kwargs()
      • LinearBasis.check_positive_flux_profile()
  • Module contents
• jaxtronomy.PointSource package
  • Subpackages
    • jaxtronomy.PointSource.Types package
      • Submodules
      • jaxtronomy.PointSource.Types.base_ps module
      • jaxtronomy.PointSource.Types.lensed_position module
      • jaxtronomy.PointSource.Types.unlensed module
      • Module contents
  • Submodules
  • jaxtronomy.PointSource.point_source module
    • PointSource
      • PointSource.__init__()
      • PointSource.update_search_window()
      • PointSource.update_lens_model()
      • PointSource.k_list()
      • PointSource.source_position()
      • PointSource.image_position()
      • PointSource.point_source_list()
      • PointSource.num_basis()
      • PointSource.image_amplitude()
      • PointSource.source_amplitude()
      • PointSource.linear_response_set()
      • PointSource.update_linear()
      • PointSource.linear_param_from_kwargs()
      • PointSource.check_image_positions()
      • PointSource.set_amplitudes()
      • PointSource.check_positive_flux()
  • Module contents
• jaxtronomy.Sampling package
  • Subpackages
    • jaxtronomy.Sampling.Likelihoods package
      • Submodules
      • jaxtronomy.Sampling.Likelihoods.image_likelihood module
      • jaxtronomy.Sampling.Likelihoods.position_likelihood module
      • Module contents
    • jaxtronomy.Sampling.Samplers package
      • Submodules
      • jaxtronomy.Sampling.Samplers.optax module
      • Module contents
  • Submodules
  • jaxtronomy.Sampling.likelihood module
    • Likelihood
      • Likelihood.__init__()
      • Likelihood.kwargs_imaging
      • Likelihood.logL()
      • Likelihood.log_likelihood()
      • Likelihood.check_bounds()
      • Likelihood.num_data
      • Likelihood.param_limits
      • Likelihood.effective_num_data_points()
      • Likelihood.likelihood()
      • Likelihood.negativelogL()
  • Module contents
• jaxtronomy.Util package
  • Submodules
  • jaxtronomy.Util.class_creator module
    • create_class_instances()
    • create_image_model()
    • create_im_sim()
  • jaxtronomy.Util.herm_util module
    • eval_hermite()
    • hermval()
  • jaxtronomy.Util.hyp2f1_util module
    • hyp2f1_series()
    • hyp2f1_continuation()
    • hyp2f1_lopez_temme_8()
    • hyp2f1_near_one()
    • hyp2f1()
  • jaxtronomy.Util.image_util module
    • add_layer2image()
    • add_layer2image_int()
    • re_size()
  • jaxtronomy.Util.kernel_util module
    • estimate_amp()
  • jaxtronomy.Util.param_util module
    • cart2polar()
    • polar2cart()
    • shear_polar2cartesian()
    • shear_cartesian2polar()
    • phi_q2_ellipticity()
    • ellipticity2phi_q()
    • transform_e1e2_product_average()
    • transform_e1e2_square_average()
    • q2e()
  • jaxtronomy.Util.util module
  • Module contents
• jaxtronomy.Workflow package
  • Submodules
  • jaxtronomy.Workflow.fitting_sequence module
    • FittingSequence
      • FittingSequence.__init__()
      • FittingSequence.kwargs_fixed
      • FittingSequence.fit_sequence()
      • FittingSequence.best_fit()
      • FittingSequence.update_state()
      • FittingSequence.best_fit_likelihood()
      • FittingSequence.bic
      • FittingSequence.param_class
      • FittingSequence.likelihood_class
      • FittingSequence.simplex()
      • FittingSequence.mcmc()
      • FittingSequence.optax()
      • FittingSequence.pso()
      • FittingSequence.nested_sampling()
      • FittingSequence.psf_iteration()
      • FittingSequence.align_images()
      • FittingSequence.update_settings()
      • FittingSequence.set_param_value()
      • FittingSequence.fix_not_computed()
      • FittingSequence.best_fit_from_samples()
      • FittingSequence.psf_iteration_memory
  • Module contents

Module contents