Index

A

antiderivative() (grater_jax.disk_model.interpolated_univariate_spline.InterpolatedUnivariateSpline method)

auto_corr() (grater_jax.optimization.mcmc_model.MCMC_model method)

C

c_to_c_osamp() (in module grater_jax.disk_model.winnie_jwst_fm) compute_phase_function_from_cosphi() (grater_jax.disk_model.SLD_utils.DoubleHenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.HenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.InterpolatedUnivariateSpline_SPF class method) compute_scattered_light_jax() (grater_jax.disk_model.SLD_ojax.ScatteredLightDisk class method) compute_stellar_psf_image() (grater_jax.disk_model.SLD_utils.LinearStellarPSF class method) (grater_jax.disk_model.SLD_utils.PositionalStellarPSF class method) (grater_jax.optimization.optimize_framework.Optimizer method)

convert_dhg_params_to_spline_params() (grater_jax.optimization.optimize_framework.OptimizeUtils class method) convolve_cube() (grater_jax.disk_model.winnie_class.WinniePSF method) convolve_with_spatial_psfs() (grater_jax.disk_model.winnie_class.WinniePSF method) (in module grater_jax.disk_model.winnie_jwst_fm) create_empirical_err_map() (grater_jax.optimization.optimize_framework.OptimizeUtils class method)

D

define_reference_images() (grater_jax.optimization.optimize_framework.Optimizer method) density_cylindrical() (grater_jax.disk_model.SLD_utils.DustEllipticalDistribution2PowerLaws class method) derivative() (grater_jax.disk_model.interpolated_univariate_spline.InterpolatedUnivariateSpline method)

disk_params (grater_jax.disk_model.objective_functions.Parameter_Index attribute), [1] dist_to_pt() (in module grater_jax.disk_model.winnie_jwst_fm) DoubleHenyeyGreenstein_SPF (class in grater_jax.disk_model.SLD_utils) DustEllipticalDistribution2PowerLaws (class in grater_jax.disk_model.SLD_utils)

E

EMP_PSF (class in grater_jax.disk_model.SLD_utils)

estimate_flux_scaling() (grater_jax.optimization.optimize_framework.Optimizer method)

F

fix_all_nonphysical_params() (grater_jax.optimization.optimize_framework.Optimizer method)

free_gpu() (in module grater_jax.disk_model.winnie_jwst_fm)

G

GAUSSIAN_PSF (class in grater_jax.disk_model.SLD_utils) generate() (grater_jax.disk_model.SLD_utils.EMP_PSF class method) (grater_jax.disk_model.SLD_utils.GAUSSIAN_PSF class method) (grater_jax.disk_model.SLD_utils.Winnie_PSF class method) generate_jwst_mask_image() (in module grater_jax.disk_model.winnie_jwst_fm) generate_nircam_psf_grid() (in module grater_jax.disk_model.winnie_jwst_fm) get_convolved_cube() (grater_jax.disk_model.winnie_class.WinniePSF method) get_disk() (grater_jax.optimization.optimize_framework.Optimizer method) get_flux_scale() (grater_jax.optimization.optimize_framework.Optimizer method) get_gradient() (grater_jax.optimization.optimize_framework.Optimizer method) get_knots() (grater_jax.disk_model.SLD_utils.InterpolatedUnivariateSpline_SPF class method) get_mask() (grater_jax.optimization.optimize_framework.OptimizeUtils class method) get_model() (grater_jax.optimization.optimize_framework.Optimizer method) get_objective_gradient() (grater_jax.optimization.optimize_framework.Optimizer method) get_objective_likelihood() (grater_jax.optimization.optimize_framework.Optimizer method) get_theta_max() (grater_jax.optimization.mcmc_model.MCMC_model method) get_theta_median() (grater_jax.optimization.mcmc_model.MCMC_model method) get_theta_percs() (grater_jax.optimization.mcmc_model.MCMC_model method) get_values() (grater_jax.optimization.optimize_framework.Optimizer method) grater_jax.disk_model module

grater_jax.disk_model.interpolated_univariate_spline module grater_jax.disk_model.jax_gradient_wrappers module grater_jax.disk_model.jax_model_wrappers module grater_jax.disk_model.objective_functions module grater_jax.disk_model.SLD_ojax module grater_jax.disk_model.SLD_utils module grater_jax.disk_model.winnie_class module grater_jax.disk_model.winnie_jwst_fm module grater_jax.optimization module grater_jax.optimization.mcmc_model module grater_jax.optimization.optimize_framework module

H

HenyeyGreenstein_SPF (class in grater_jax.disk_model.SLD_utils)

I

im_mask_rolls (grater_jax.disk_model.winnie_class.WinniePSF attribute) img (grater_jax.disk_model.SLD_utils.EMP_PSF attribute) inc_bound_knots() (grater_jax.optimization.optimize_framework.Optimizer method) init() (grater_jax.disk_model.SLD_ojax.ScatteredLightDisk class method) (grater_jax.disk_model.SLD_utils.DoubleHenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.DustEllipticalDistribution2PowerLaws class method) (grater_jax.disk_model.SLD_utils.HenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.InterpolatedUnivariateSpline_SPF class method) (grater_jax.disk_model.SLD_utils.Winnie_PSF class method)

init_db() (grater_jax.disk_model.winnie_class.WinniePSF class method) initialize_stellar_psf_weights() (grater_jax.optimization.optimize_framework.Optimizer method) integral() (grater_jax.disk_model.interpolated_univariate_spline.InterpolatedUnivariateSpline method) InterpolatedUnivariateSpline (class in grater_jax.disk_model.interpolated_univariate_spline) InterpolatedUnivariateSpline_SPF (class in grater_jax.disk_model.SLD_utils) isNone() (in module grater_jax.disk_model.winnie_jwst_fm)

J

Jax_class (class in grater_jax.disk_model.SLD_utils) jax_model() (in module grater_jax.disk_model.jax_model_wrappers) jax_model_grad() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_ll() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_spline() (in module grater_jax.disk_model.jax_model_wrappers) jax_model_spline_grad() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_spline_ll() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_spline_winnie() (in module grater_jax.disk_model.jax_model_wrappers)

jax_model_spline_winnie_grad() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_spline_winnie_ll() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_winnie() (in module grater_jax.disk_model.jax_model_wrappers) jax_model_winnie_grad() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_model_winnie_ll() (in module grater_jax.disk_model.jax_gradient_wrappers) jax_rotate_image() (in module grater_jax.disk_model.winnie_class) jit_compile_gradient() (grater_jax.optimization.optimize_framework.Optimizer method) jit_compile_model() (grater_jax.optimization.optimize_framework.Optimizer method)

L

LinearStellarPSF (class in grater_jax.disk_model.SLD_utils) load_machine_readable() (grater_jax.optimization.optimize_framework.Optimizer method)

log_likelihood() (grater_jax.optimization.optimize_framework.Optimizer method) (in module grater_jax.disk_model.jax_gradient_wrappers) log_likelihood_pos() (grater_jax.optimization.optimize_framework.Optimizer method)

M

mcmc() (grater_jax.optimization.optimize_framework.Optimizer method) MCMC_model (class in grater_jax.optimization.mcmc_model) misc_params (grater_jax.disk_model.objective_functions.Parameter_Index attribute), [1] module grater_jax.disk_model grater_jax.disk_model.interpolated_univariate_spline grater_jax.disk_model.jax_gradient_wrappers grater_jax.disk_model.jax_model_wrappers grater_jax.disk_model.objective_functions grater_jax.disk_model.SLD_ojax grater_jax.disk_model.SLD_utils grater_jax.disk_model.winnie_class grater_jax.disk_model.winnie_jwst_fm grater_jax.optimization grater_jax.optimization.mcmc_model grater_jax.optimization.optimize_framework

mpl_stcen_extent() (in module grater_jax.disk_model.winnie_jwst_fm)

N

num_unique_psfs (grater_jax.disk_model.winnie_class.WinniePSF attribute)

O

objective_fit() (in module grater_jax.disk_model.objective_functions) objective_fit_grad() (in module grater_jax.disk_model.objective_functions) objective_grad() (in module grater_jax.disk_model.objective_functions)

objective_ll() (in module grater_jax.disk_model.objective_functions) objective_model() (in module grater_jax.disk_model.objective_functions) Optimizer (class in grater_jax.optimization.optimize_framework) OptimizeUtils (class in grater_jax.optimization.optimize_framework)

P

pack_pars() (grater_jax.disk_model.SLD_ojax.ScatteredLightDisk class method) (grater_jax.disk_model.SLD_utils.DoubleHenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.DustEllipticalDistribution2PowerLaws class method) (grater_jax.disk_model.SLD_utils.EMP_PSF class method) (grater_jax.disk_model.SLD_utils.GAUSSIAN_PSF class method) (grater_jax.disk_model.SLD_utils.HenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.InterpolatedUnivariateSpline_SPF class method) (grater_jax.disk_model.SLD_utils.Jax_class class method) (grater_jax.disk_model.SLD_utils.LinearStellarPSF class method) (grater_jax.disk_model.SLD_utils.PositionalStellarPSF class method) (grater_jax.disk_model.SLD_utils.Winnie_PSF class method) (in module grater_jax.disk_model.objective_functions) pad_and_rotate_hypercube() (in module grater_jax.disk_model.winnie_jwst_fm) pad_or_crop_about_pos() (in module grater_jax.disk_model.winnie_jwst_fm) pad_or_crop_image() (in module grater_jax.disk_model.winnie_jwst_fm) Parameter_Index (class in grater_jax.disk_model.objective_functions) params (grater_jax.disk_model.SLD_ojax.ScatteredLightDisk attribute) (grater_jax.disk_model.SLD_utils.DoubleHenyeyGreenstein_SPF attribute) (grater_jax.disk_model.SLD_utils.DustEllipticalDistribution2PowerLaws attribute) (grater_jax.disk_model.SLD_utils.EMP_PSF attribute) (grater_jax.disk_model.SLD_utils.GAUSSIAN_PSF attribute) (grater_jax.disk_model.SLD_utils.HenyeyGreenstein_SPF attribute) (grater_jax.disk_model.SLD_utils.InterpolatedUnivariateSpline_SPF attribute) (grater_jax.disk_model.SLD_utils.Jax_class attribute) (grater_jax.disk_model.SLD_utils.LinearStellarPSF attribute) (grater_jax.disk_model.SLD_utils.PositionalStellarPSF attribute) (grater_jax.disk_model.SLD_utils.Winnie_PSF attribute)

plot_chains() (grater_jax.optimization.mcmc_model.MCMC_model method) plot_spline() (grater_jax.optimization.optimize_framework.Optimizer method) PositionalStellarPSF (class in grater_jax.disk_model.SLD_utils) print_params() (grater_jax.optimization.optimize_framework.Optimizer method) process_image() (grater_jax.optimization.optimize_framework.OptimizeUtils class method) propagate_nans_in_spatial_operation() (in module grater_jax.disk_model.winnie_jwst_fm) psf_convolve_cpu() (in module grater_jax.disk_model.winnie_jwst_fm) psf_convolve_gpu() (in module grater_jax.disk_model.winnie_jwst_fm) psf_inds_rolls (grater_jax.disk_model.winnie_class.WinniePSF attribute) psf_offsets (grater_jax.disk_model.winnie_class.WinniePSF attribute) psf_parangs (grater_jax.disk_model.winnie_class.WinniePSF attribute) psfs (grater_jax.disk_model.winnie_class.WinniePSF attribute)

R

raw_model_to_convolved_model_cube() (in module grater_jax.disk_model.winnie_jwst_fm) recommended_num_knots() (in module grater_jax.disk_model.SLD_utils) reference_images (grater_jax.disk_model.SLD_utils.StellarPSFReference attribute) residuals() (in module grater_jax.disk_model.jax_gradient_wrappers) rotate_about_pos() (in module grater_jax.disk_model.winnie_jwst_fm)

rotate_about_pos_gpu() (in module grater_jax.disk_model.winnie_jwst_fm) rotate_hypercube() (grater_jax.disk_model.winnie_class.WinniePSF method) (in module grater_jax.disk_model.winnie_jwst_fm) rotate_image_cpu() (in module grater_jax.disk_model.winnie_jwst_fm) rotate_image_gpu() (in module grater_jax.disk_model.winnie_jwst_fm) run() (grater_jax.optimization.mcmc_model.MCMC_model method)

S

save_human_readable() (grater_jax.optimization.optimize_framework.Optimizer method) save_machine_readable() (grater_jax.optimization.optimize_framework.Optimizer method) scale_spline_chains() (grater_jax.optimization.optimize_framework.OptimizeUtils class method) ScatteredLightDisk (class in grater_jax.disk_model.SLD_ojax) scipy_bounded_optimize() (grater_jax.optimization.optimize_framework.Optimizer method)

scipy_optimize() (grater_jax.optimization.optimize_framework.Optimizer method) set_discarded_iters() (grater_jax.optimization.mcmc_model.MCMC_model method) set_empirical_psf() (grater_jax.optimization.optimize_framework.Optimizer method) set_throughput() (grater_jax.optimization.optimize_framework.Optimizer method) show_corner_plot() (grater_jax.optimization.mcmc_model.MCMC_model method) StellarPSFReference (class in grater_jax.disk_model.SLD_utils)

T

tree_flatten() (grater_jax.disk_model.interpolated_univariate_spline.InterpolatedUnivariateSpline method) (grater_jax.disk_model.winnie_class.WinniePSF method)

tree_unflatten() (grater_jax.disk_model.interpolated_univariate_spline.InterpolatedUnivariateSpline class method) (grater_jax.disk_model.winnie_class.WinniePSF class method)

U

unlogscale_mcmc_model() (grater_jax.optimization.optimize_framework.OptimizeUtils class method) unpack_pars() (grater_jax.disk_model.SLD_ojax.ScatteredLightDisk class method) (grater_jax.disk_model.SLD_utils.DoubleHenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.DustEllipticalDistribution2PowerLaws class method) (grater_jax.disk_model.SLD_utils.EMP_PSF class method) (grater_jax.disk_model.SLD_utils.GAUSSIAN_PSF class method) (grater_jax.disk_model.SLD_utils.HenyeyGreenstein_SPF class method) (grater_jax.disk_model.SLD_utils.InterpolatedUnivariateSpline_SPF class method) (grater_jax.disk_model.SLD_utils.Jax_class class method) (grater_jax.disk_model.SLD_utils.LinearStellarPSF class method) (grater_jax.disk_model.SLD_utils.PositionalStellarPSF class method) (grater_jax.disk_model.SLD_utils.Winnie_PSF class method)

W

warm_start_knots() (grater_jax.optimization.optimize_framework.Optimizer method)

Winnie_PSF (class in grater_jax.disk_model.SLD_utils) WinniePSF (class in grater_jax.disk_model.winnie_class)

X

xy_polar_ang_displacement() (in module grater_jax.disk_model.winnie_jwst_fm)

xy_polar_ang_displacement_gpu() (in module grater_jax.disk_model.winnie_jwst_fm)