1.1. autojax package¶
1.1.1. Subpackages¶
1.1.2. Submodules¶
- 1.1.2.1. autojax.jax module
constant_regularization_matrix_from()curvature_matrix_via_w_compact_sparse_mapping_matrix_from()curvature_matrix_via_w_compact_sparse_mapping_matrix_from_BCOO()curvature_matrix_via_w_tilde_curvature_preload_interferometer_from()curvature_matrix_via_w_tilde_from()curvature_matrix_via_w_wilde_sparse_mapping_matrix_from()data_vector_from()log_likelihood_function_via_w_compact_from()log_likelihood_function_via_w_tilde_from()mapping_matrix_from()mapping_matrix_from_BCOO()mask_2d_centres_from()mask_2d_circular_from()noise_normalization_complex_from()reconstruction_positive_negative_from()sparse_mapping_matrix_transpose_matmul()w_compact_curvature_interferometer_from()w_tilde_curvature_interferometer_from()w_tilde_data_interferometer_from()w_tilde_via_compact_from()w_tilde_via_preload_from()
- 1.1.2.2. autojax.numba module
constant_regularization_matrix_from()curvature_matrix_via_w_compact_sparse_mapping_matrix_direct_from()curvature_matrix_via_w_compact_sparse_mapping_matrix_from()curvature_matrix_via_w_tilde_curvature_preload_interferometer_from()curvature_matrix_via_w_tilde_from()curvature_matrix_via_w_wilde_sparse_mapping_matrix_from()data_vector_from()log_likelihood_function_via_w_compact_from()log_likelihood_function_via_w_tilde_from()mapping_matrix_from()mask_2d_centres_from()mask_2d_circular_from()noise_normalization_complex_from()reconstruction_positive_negative_from()sparse_mapping_matrix_transpose_matmul()w_compact_curvature_interferometer_from()w_tilde_curvature_interferometer_from()w_tilde_data_interferometer_from()w_tilde_via_compact_from()w_tilde_via_preload_from()
- 1.1.2.3. autojax.original module
constant_regularization_matrix_from()curvature_matrix_via_w_tilde_curvature_preload_interferometer_from()curvature_matrix_via_w_tilde_from()data_vector_from()log_likelihood_function_via_w_compact_from()log_likelihood_function_via_w_tilde_from()mapping_matrix_from()mask_2d_centres_from()mask_2d_circular_from()noise_normalization_complex_from()reconstruction_positive_negative_from()w_tilde_curvature_interferometer_from()w_tilde_curvature_preload_interferometer_from()w_tilde_data_interferometer_from()w_tilde_via_preload_from()
- 1.1.2.4. autojax.tests module
AutoTestMetaDataData.BData.KData.MData.NData.N_Data.N_PRIMEData.PData.SData.centreData.coefficientData.curvature_matrixData.curvature_reg_matrixData.dataData.data_vectorData.dict()Data.dirty_imageData.grid_radians_2dData.grid_radians_slimData.mapping_matrixData.native_index_for_slim_indexData.neighborsData.neighbors_gridData.neighbors_sizesData.noise_mapData.noise_map_realData.pix_indexes_for_sub_slim_indexData.pix_pixelsData.pix_size_for_sub_slim_indexData.pix_weights_for_sub_slim_indexData.pixel_scaleData.pixel_scalesData.radiusData.real_space_maskData.regularization_matrixData.shape_masked_pixels_2dData.shape_nativeData.slim_index_for_sub_slim_indexData.sub_fractionData.uv_wavelengthsData.visibilities_realData.w_compactData.w_tildeData.w_tilde_preload
DataGeneratedDataGenerated.KDataGenerated.K_DataGenerated.PDataGenerated.P_DataGenerated.SDataGenerated.S_DataGenerated.dataDataGenerated.dirty_imageDataGenerated.neighborsDataGenerated.neighbors_sizesDataGenerated.noise_mapDataGenerated.pix_indexes_for_sub_slim_indexDataGenerated.pix_weights_for_sub_slim_indexDataGenerated.uv_wavelengths
DataLoadedReferenceTestCurvatureMatrixTestCurvatureMatrix.test_curvature_matrix_jax()TestCurvatureMatrix.test_curvature_matrix_jax_BCOO()TestCurvatureMatrix.test_curvature_matrix_jax_compact_sparse()TestCurvatureMatrix.test_curvature_matrix_jax_compact_sparse_BCOO()TestCurvatureMatrix.test_curvature_matrix_jax_sparse()TestCurvatureMatrix.test_curvature_matrix_numba()TestCurvatureMatrix.test_curvature_matrix_numba_compact_sparse()TestCurvatureMatrix.test_curvature_matrix_numba_compact_sparse_direct()TestCurvatureMatrix.test_curvature_matrix_numba_sparse()TestCurvatureMatrix.test_curvature_matrix_original()TestCurvatureMatrix.test_curvature_matrix_original_preload_direct()
TestJaxTestJax.modTestJax.test_constant_regularization_matrix_from_jax()TestJax.test_curvature_matrix_via_w_compact_sparse_mapping_matrix_from_jax()TestJax.test_curvature_matrix_via_w_tilde_curvature_preload_interferometer_from_jax()TestJax.test_curvature_matrix_via_w_tilde_from_jax()TestJax.test_data_vector_from_jax()TestJax.test_log_likelihood_function_via_w_compact_from_jax()TestJax.test_log_likelihood_function_via_w_tilde_from_jax()TestJax.test_mapping_matrix_from_jax()TestJax.test_mask_2d_circular_from_jax()TestJax.test_noise_normalization_complex_from_jax()TestJax.test_reconstruction_positive_negative_from_jax()TestJax.test_sparse_mapping_matrix_transpose_matmul_jax()TestJax.test_w_compact_curvature_interferometer_from_jax()TestJax.test_w_tilde_curvature_interferometer_from_jax()TestJax.test_w_tilde_curvature_preload_interferometer_from_jax()TestJax.test_w_tilde_data_interferometer_from_jax()TestJax.test_w_tilde_via_compact_from_jax()TestJax.test_w_tilde_via_preload_from_jax()
TestNumbaTestNumba.modTestNumba.test_constant_regularization_matrix_from_numba()TestNumba.test_curvature_matrix_via_w_compact_sparse_mapping_matrix_from_numba()TestNumba.test_curvature_matrix_via_w_tilde_curvature_preload_interferometer_from_numba()TestNumba.test_curvature_matrix_via_w_tilde_from_numba()TestNumba.test_data_vector_from_numba()TestNumba.test_log_likelihood_function_via_w_compact_from_numba()TestNumba.test_log_likelihood_function_via_w_tilde_from_numba()TestNumba.test_mapping_matrix_from_numba()TestNumba.test_mask_2d_circular_from_numba()TestNumba.test_noise_normalization_complex_from_numba()TestNumba.test_reconstruction_positive_negative_from_numba()TestNumba.test_sparse_mapping_matrix_transpose_matmul_numba()TestNumba.test_w_compact_curvature_interferometer_from_numba()TestNumba.test_w_tilde_curvature_interferometer_from_numba()TestNumba.test_w_tilde_curvature_preload_interferometer_from_numba()TestNumba.test_w_tilde_data_interferometer_from_numba()TestNumba.test_w_tilde_via_compact_from_numba()TestNumba.test_w_tilde_via_preload_from_numba()
TestOriginalTestOriginal.modTestOriginal.test_constant_regularization_matrix_from_original()TestOriginal.test_curvature_matrix_via_w_compact_sparse_mapping_matrix_from_original()TestOriginal.test_curvature_matrix_via_w_tilde_curvature_preload_interferometer_from_original()TestOriginal.test_curvature_matrix_via_w_tilde_from_original()TestOriginal.test_data_vector_from_original()TestOriginal.test_log_likelihood_function_via_w_compact_from_original()TestOriginal.test_log_likelihood_function_via_w_tilde_from_original()TestOriginal.test_mapping_matrix_from_original()TestOriginal.test_mask_2d_circular_from_original()TestOriginal.test_noise_normalization_complex_from_original()TestOriginal.test_reconstruction_positive_negative_from_original()TestOriginal.test_sparse_mapping_matrix_transpose_matmul_original()TestOriginal.test_w_compact_curvature_interferometer_from_original()TestOriginal.test_w_tilde_curvature_interferometer_from_original()TestOriginal.test_w_tilde_curvature_preload_interferometer_from_original()TestOriginal.test_w_tilde_data_interferometer_from_original()TestOriginal.test_w_tilde_via_compact_from_original()TestOriginal.test_w_tilde_via_preload_from_original()
TestWTildeTestWTilde.test_w_tilde_curvature_interferometer_from_jax_compact()TestWTilde.test_w_tilde_curvature_interferometer_from_jax_compact_expanded()TestWTilde.test_w_tilde_curvature_interferometer_from_numba_compact()TestWTilde.test_w_tilde_curvature_interferometer_from_numba_compact_expanded()TestWTilde.test_w_tilde_curvature_interferometer_from_original_preload()TestWTilde.test_w_tilde_curvature_interferometer_from_original_preload_expanded()
data_bundle()deterministic_seed()gen_neighbors()gen_pix_indexes_for_sub_slim_index()get_run()get_run_composed_from()get_run_composed_from_prepend()neighbors_grid()