Index

_

__init__() (jammy_flows.amortizable_mlp.AmortizableMLP method) (jammy_flows.layers.euclidean.euclidean_base.euclidean_base method) (jammy_flows.layers.euclidean.euclidean_do_nothing.euclidean_do_nothing method) (jammy_flows.layers.euclidean.gaussianization_flow.gf_block method) (jammy_flows.layers.euclidean.multivariate_normal.mvn_block method) (jammy_flows.layers.euclidean.polynomial_stretch_flow.psf_block method) (jammy_flows.layers.intervals.interval_base.interval_base method) (jammy_flows.layers.intervals.interval_do_nothing.interval_do_nothing method) (jammy_flows.layers.intervals.rational_quadratic_spline.rational_quadratic_spline method) (jammy_flows.layers.layer_base.layer_base method) (jammy_flows.layers.simplex.inner_loop_simplex.inner_loop_simplex method) (jammy_flows.layers.simplex.simplex_base.simplex_base method) (jammy_flows.layers.spheres.cnf_sphere_charts.cnf_sphere_charts method) (jammy_flows.layers.spheres.exponential_map_s2.exponential_map_s2 method) (jammy_flows.layers.spheres.moebius_1d.moebius method) (jammy_flows.layers.spheres.segmented_sphere_nd.segmented_sphere_nd method) (jammy_flows.layers.spheres.sphere_base.sphere_base method) (jammy_flows.layers.spheres.spherical_do_nothing.spherical_do_nothing method) (jammy_flows.layers.spheres.splines_1d.spline_1d method)

_embedding_conditional_return() (jammy_flows.layers.layer_base.layer_base method) _embedding_conditional_return_num() (jammy_flows.layers.layer_base.layer_base method)

A

AmortizableMLP (class in jammy_flows.amortizable_mlp)

C

cnf_sphere_charts (class in jammy_flows.layers.spheres.cnf_sphere_charts)

E

euclidean_base (class in jammy_flows.layers.euclidean.euclidean_base)

euclidean_do_nothing (class in jammy_flows.layers.euclidean.euclidean_do_nothing) exponential_map_s2 (class in jammy_flows.layers.spheres.exponential_map_s2)

F

flow_mapping() (jammy_flows.layers.layer_base.layer_base method)

forward() (jammy_flows.amortizable_mlp.AmortizableMLP method)

G

get_desired_init_parameters() (jammy_flows.layers.layer_base.layer_base method)

gf_block (class in jammy_flows.layers.euclidean.gaussianization_flow)

I

init_params() (jammy_flows.layers.layer_base.layer_base method) initialize_uvbs() (jammy_flows.amortizable_mlp.AmortizableMLP method) inner_loop_simplex (class in jammy_flows.layers.simplex.inner_loop_simplex) interval_base (class in jammy_flows.layers.intervals.interval_base) interval_do_nothing (class in jammy_flows.layers.intervals.interval_do_nothing)

inv_flow_mapping() (jammy_flows.layers.layer_base.layer_base method) inverse_bisection_n_newton() (in module jammy_flows.layers.bisection_n_newton) inverse_bisection_n_newton_joint_func_and_grad() (in module jammy_flows.layers.bisection_n_newton) inverse_bisection_n_newton_sphere() (in module jammy_flows.layers.bisection_n_newton) inverse_bisection_n_newton_sphere_fast() (in module jammy_flows.layers.bisection_n_newton)

J

jammy_flows.amortizable_mlp (module) jammy_flows.extra_functions (module) jammy_flows.flow_options (module) jammy_flows.helper_fns (module) jammy_flows.layers.bisection_n_newton (module) jammy_flows.layers.euclidean.euclidean_base (module) jammy_flows.layers.euclidean.euclidean_do_nothing (module) jammy_flows.layers.euclidean.gaussianization_flow (module) jammy_flows.layers.euclidean.multivariate_normal (module) jammy_flows.layers.euclidean.polynomial_stretch_flow (module) jammy_flows.layers.intervals.interval_base (module)

jammy_flows.layers.intervals.interval_do_nothing (module) jammy_flows.layers.intervals.rational_quadratic_spline (module) jammy_flows.layers.layer_base (module) jammy_flows.layers.simplex.inner_loop_simplex (module) jammy_flows.layers.simplex.simplex_base (module) jammy_flows.layers.spheres.cnf_sphere_charts (module) jammy_flows.layers.spheres.exponential_map_s2 (module) jammy_flows.layers.spheres.moebius_1d (module) jammy_flows.layers.spheres.segmented_sphere_nd (module) jammy_flows.layers.spheres.sphere_base (module) jammy_flows.layers.spheres.spherical_do_nothing (module) jammy_flows.layers.spheres.splines_1d (module)

L

layer_base (class in jammy_flows.layers.layer_base)

log_one_plus_exp_x_to_a_minus_1() (in module jammy_flows.extra_functions)

M

moebius (class in jammy_flows.layers.spheres.moebius_1d)

mvn_block (class in jammy_flows.layers.euclidean.multivariate_normal)

N

num_parameter_list (jammy_flows.layers.spheres.segmented_sphere_nd.segmented_sphere_nd attribute)

O

obtain_default_init_tensor() (jammy_flows.amortizable_mlp.AmortizableMLP method)

obtain_default_options() (in module jammy_flows.flow_options) obtain_layer_param_structure() (jammy_flows.layers.layer_base.layer_base method)

P

psf_block (class in jammy_flows.layers.euclidean.polynomial_stretch_flow)

R

rational_quadratic_spline (class in jammy_flows.layers.intervals.rational_quadratic_spline)

return_safe_costheta() (in module jammy_flows.layers.spheres.sphere_base)

S

segmented_sphere_nd (class in jammy_flows.layers.spheres.segmented_sphere_nd) simplex_base (class in jammy_flows.layers.simplex.simplex_base)

sphere_base (class in jammy_flows.layers.spheres.sphere_base) spherical_do_nothing (class in jammy_flows.layers.spheres.spherical_do_nothing) spline_1d (class in jammy_flows.layers.spheres.splines_1d)

T

transform_target_space() (jammy_flows.layers.layer_base.layer_base method)