Search.setIndex({docnames:["index","pages/details","pages/examples","pages/examples/2d_field_inference/2d_field_inference","pages/examples/additive_noise/additive_noise","pages/examples/imnn_vs_regression/imnn_vs_regression","pages/examples/lfi","pages/examples/lfi/mean_and_variance/mean_and_variance","pages/examples/subclasses","pages/examples/subclasses/AggregatedGradientIMNN/AggregatedGradientIMNN","pages/examples/subclasses/AggregatedNumericalGradientIMNN/AggregatedNumericalGradientIMNN","pages/examples/subclasses/AggregatedSimulatorIMNN/AggregatedSimulatorIMNN","pages/examples/subclasses/DatasetGradientIMNN/DatasetGradientIMNN","pages/examples/subclasses/DatasetNumericalGradientIMNN/DatasetNumericalGradientIMNN","pages/examples/subclasses/GradientIMNN/GradientIMNN","pages/examples/subclasses/NumericalGradientIMNN/NumericalGradientIMNN","pages/examples/subclasses/SimulatorIMNN/SimulatorIMNN","pages/install","pages/lfi","pages/modules","pages/parents","pages/tf.data.Datasets"],envversion:{"sphinx.domains.c":2,"sphinx.domains.changeset":1,"sphinx.domains.citation":1,"sphinx.domains.cpp":3,"sphinx.domains.index":1,"sphinx.domains.javascript":2,"sphinx.domains.math":2,"sphinx.domains.python":2,"sphinx.domains.rst":2,"sphinx.domains.std":2,"sphinx.ext.intersphinx":1,"sphinx.ext.todo":2,sphinx:56},filenames:["index.rst","pages/details.rst","pages/examples.rst","pages/examples/2d_field_inference/2d_field_inference.rst","pages/examples/additive_noise/additive_noise.rst","pages/examples/imnn_vs_regression/imnn_vs_regression.rst","pages/examples/lfi.rst","pages/examples/lfi/mean_and_variance/mean_and_variance.rst","pages/examples/subclasses.rst","pages/examples/subclasses/AggregatedGradientIMNN/AggregatedGradientIMNN.rst","pages/examples/subclasses/AggregatedNumericalGradientIMNN/AggregatedNumericalGradientIMNN.rst","pages/examples/subclasses/AggregatedSimulatorIMNN/AggregatedSimulatorIMNN.rst","pages/examples/subclasses/DatasetGradientIMNN/DatasetGradientIMNN.rst","pages/examples/subclasses/DatasetNumericalGradientIMNN/DatasetNumericalGradientIMNN.rst","pages/examples/subclasses/GradientIMNN/GradientIMNN.rst","pages/examples/subclasses/NumericalGradientIMNN/NumericalGradientIMNN.rst","pages/examples/subclasses/SimulatorIMNN/SimulatorIMNN.rst","pages/install.rst","pages/lfi.rst","pages/modules.rst","pages/parents.rst","pages/tf.data.Datasets.rst"],objects:{"imnn.AggregatedGradientIMNN":{"_split_d\u039b_dx":[19,1,1,"imnn.AggregatedGradientIMNN._split_d_dx"],_collect_input:[19,1,1,""],_set_dataset:[19,1,1,""],_set_shapes:[19,1,1,""],get_gradient:[19,1,1,""],get_summaries:[19,1,1,""],get_summary:[19,1,1,""]},"imnn.AggregatedNumericalGradientIMNN":{"_split_d\u039b_dx":[19,1,1,"imnn.AggregatedNumericalGradientIMNN._split_d_dx"],_collect_input:[19,1,1,""],_set_batch_functions:[19,1,1,""],_set_dataset:[19,1,1,""],_set_shapes:[19,1,1,""],get_gradient:[19,1,1,""],get_summaries:[19,1,1,""],get_summary:[19,1,1,""]},"imnn.AggregatedSimulatorIMNN":{"_split_d\u039b_dx":[19,1,1,"imnn.AggregatedSimulatorIMNN._split_d_dx"],_collect_input:[19,1,1,""],_set_shapes:[19,1,1,""],get_gradient:[19,1,1,""],get_summaries:[19,1,1,""],get_summary:[19,1,1,""]},"imnn.DatasetGradientIMNN":{_set_data:[21,1,1,""],_set_dataset:[21,1,1,""],_set_prebuilt_dataset:[21,1,1,""]},"imnn.DatasetNumericalGradientIMNN":{_set_data:[21,1,1,""],_set_dataset:[21,1,1,""]},"imnn.GradientIMNN":{_set_data:[19,1,1,""],get_summaries:[19,1,1,""]},"imnn.NumericalGradientIMNN":{_collect_input:[19,1,1,""],_construct_derivatives:[19,1,1,""],_set_data:[19,1,1,""],get_summaries:[19,1,1,""]},"imnn.SimulatorIMNN":{_get_fitting_keys:[19,1,1,""],get_summaries:[19,1,1,""]},"imnn.TFRecords":{_bytes_feature:[21,1,1,""],_int64_feature:[21,1,1,""],check_func:[21,1,1,""],check_params:[21,1,1,""],check_size:[21,1,1,""],derivative_parser:[21,1,1,""],derivative_serialiser:[21,1,1,""],fiducial_serialiser:[21,1,1,""],get_file:[21,1,1,""],get_initial_seed:[21,1,1,""],get_seed:[21,1,1,""],get_serialiser:[21,1,1,""],numerical_derivative_parser:[21,1,1,""],parser:[21,1,1,""],write_record:[21,1,1,""]},"imnn.imnn._aggregated_imnn":{_AggregatedIMNN:[20,0,1,""]},"imnn.imnn._aggregated_imnn._AggregatedIMNN":{"_split_d\u039b_dx":[20,1,1,"imnn.imnn._aggregated_imnn._AggregatedIMNN._split_d_dx"],_collect_input:[20,1,1,""],_construct_gradient:[20,1,1,""],_get_batch_summaries:[20,1,1,""],_set_batch_functions:[20,1,1,""],_set_devices:[20,1,1,""],_set_shapes:[20,1,1,""],_setup_progress_bar:[20,1,1,""],_update_progress_bar:[20,1,1,""],fit:[20,1,1,""],get_gradient:[20,1,1,""],get_summaries:[20,1,1,""]},"imnn.imnn._imnn":{_IMNN:[20,0,1,""]},"imnn.imnn._imnn._IMNN":{"get_\u03b1":[20,1,1,"imnn.imnn._imnn._IMNN.get_"],_calculate_F_statistics:[20,1,1,""],_calculate_loss:[20,1,1,""],_check_loop_vars:[20,1,1,""],_construct_derivatives:[20,1,1,""],_fit:[20,1,1,""],_fit_cond:[20,1,1,""],_get_F_statistics:[20,1,1,""],_get_fitting_keys:[20,1,1,""],_get_loss:[20,1,1,""],_get_regularisation:[20,1,1,""],_get_regularisation_strength:[20,1,1,""],_initialise_history:[20,1,1,""],_initialise_model:[20,1,1,""],_initialise_parameters:[20,1,1,""],_set_history:[20,1,1,""],_set_inputs:[20,1,1,""],_setup_plot:[20,1,1,""],_slogdet:[20,1,1,""],_update_history:[20,1,1,""],_update_loop_vars:[20,1,1,""],fit:[20,1,1,""],get_estimate:[20,1,1,""],get_summaries:[20,1,1,""],plot:[20,1,1,""],set_F_statistics:[20,1,1,""]},"imnn.lfi":{ApproximateBayesianComputation:[18,0,1,""],GaussianApproximation:[18,0,1,""],LikelihoodFreeInference:[18,0,1,""],PopulationMonteCarlo:[18,0,1,""]},"imnn.lfi.ApproximateBayesianComputation":{F_distance:[18,1,1,""],compressor:[18,1,1,""],euclidean_distance:[18,1,1,""],get_marginals:[18,1,1,""],get_min_accepted:[18,1,1,""],get_samples:[18,1,1,""],scatter_plot:[18,1,1,""],scatter_summaries:[18,1,1,""],set_accepted:[18,1,1,""],set_marginals:[18,1,1,""],set_samples:[18,1,1,""],simulator:[18,1,1,""]},"imnn.lfi.GaussianApproximation":{get_marginals:[18,1,1,""]},"imnn.lfi.LikelihoodFreeInference":{_scatter_plot:[18,1,1,""],get_gridsize:[18,1,1,""],get_levels:[18,1,1,""],marginal_plot:[18,1,1,""],prior:[18,1,1,""],put_marginals:[18,1,1,""],setup_plot:[18,1,1,""],target_choice:[18,1,1,""]},"imnn.lfi.PopulationMonteCarlo":{move_samples:[18,1,1,""],set_accepted:[18,1,1,""],set_samples:[18,1,1,""]},imnn:{AggregatedGradientIMNN:[19,0,1,""],AggregatedNumericalGradientIMNN:[19,0,1,""],AggregatedSimulatorIMNN:[19,0,1,""],DatasetGradientIMNN:[21,0,1,""],DatasetNumericalGradientIMNN:[21,0,1,""],GradientIMNN:[19,0,1,""],IMNN:[2,2,1,""],NumericalGradientIMNN:[19,0,1,""],SimulatorIMNN:[19,0,1,""],TFRecords:[21,0,1,""]}},objnames:{"0":["py","class","Python class"],"1":["py","method","Python method"],"2":["py","function","Python function"]},objtypes:{"0":"py:class","1":"py:method","2":"py:function"},terms:{"001":[4,9,10,11,12,13,14,15,16],"0029":1,"0084834":5,"011085":7,"020965":20,"04d":[12,13,21],"06379524":3,"071932":5,"076218":20,"0789695":20,"0823894":20,"083004":1,"08444":7,"0it":[4,5,11,14,15,16],"100":[3,4,5,7,9,10,11,12,13,14,15,16,18,20,21],"1000":[3,4,5,7,9,10,11,12,13,14,15,16,20,21],"10000":[3,7],"100000":[5,14,15,16,20],"1004":5,"1008":7,"101":5,"1010":3,"1043":5,"10615":7,"10693721":20,"1103":1,"1108716":20,"1113386":20,"114669":20,"128":[3,4,5,9,10,11,12,13,14,15,16,20],"128591":20,"13min":3,"1400":3,"14554":20,"14min":3,"150":21,"1545":3,"1600000":7,"1690":3,"1901293":20,"1956":5,"1984":3,"19min":3,"1e5":[7,18,20],"200":[5,7,12,13,21],"2000":4,"2003":5,"2018":1,"2021":1,"2034":5,"21s":3,"229679":20,"23min":3,"2470":1,"248898":20,"250":[3,20],"2500000":4,"2622285":20,"2800000":5,"2819":5,"2_a":3,"2_b":3,"31980422":20,"337":7,"34187225":7,"344688":20,"38s":3,"398947":20,"3min":3,"40224":5,"40968":7,"4134324":20,"430000":3,"47132865":20,"472563":20,"4816203":5,"482665":20,"4837523":20,"48s":3,"49264978":3,"500":[3,4],"5000":[3,4],"5172062":20,"5321330648":3,"54s":3,"5737753":20,"5812075":20,"6041985":20,"611":5,"611000":5,"626409":20,"6470118":20,"659824":7,"668312":20,"6701807":7,"6727388":20,"70872":20,"7216978":7,"7219954":20,"734651":20,"7369494":20,"745726":20,"7684307":20,"7861238":20,"7881424":20,"8083427":20,"828809":20,"83408386":5,"856505":20,"8727236":20,"87299":20,"894459":20,"900367":20,"95049":7,"95406663":7,"967560":3,"97123594":3,"978279":20,"982314":7,"98241055":5,"9941278":5,"\u03b1":20,"\u03b4":3,"\u03b4\u03b8":[1,2,4,10,13,15,19,21],"\u03b4_target":3,"\u03b4m":4,"\u03b5":[18,20],"\u03b5_ind":18,"\u03b6":4,"\u03b8":[3,4,7,9,10,11,12,13,14,15,16,19,20,21],"\u03b8_der":[10,13,15,21],"\u03b8_fid":[1,2,3,4,5,7,9,10,11,12,13,14,15,16,19,20,21],"\u03b8_target":3,"\u03bb":[1,3,4,5,9,10,11,12,13,14,15,16,20],"\u03bb2":20,"\u03bc":[7,20],"\u03c3":[7,20],"\u03f5":[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20],"\u03f5_ind":18,"\ud835\udf06":4,"\ud835\udf16":4,"_\u03bb2":20,"_split_d\u03bb_dx":[19,20],"boolean":18,"break":[18,21],"byte":21,"case":[1,3,4,7,18,19,20,21],"class":[0,1,4,5,8,12,13,18,21],"d2\u03bb_dxd\u03b8":[19,20],"d2x_dwd\u03b8":20,"d\u03bb_dx":[4,19,20],"d\u03bc_d\u03b8":20,"dd_d\u03b8":[19,20],"default":[2,14,15,16,18,19,20,21],"final":[1,3,9,10,11,12,13,14,15,16,18,20,21],"float":[1,2,18,19,20,21],"function":[1,2,3,4,5,7,10,11,12,13,14,15,16,17,18,19,20,21],"get_\u03b1":20,"import":[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20,21],"int":[1,2,3,7,18,19,20,21],"long":[1,12,13,18,19,20,21],"new":[1,2,3,7,16,18,19,20],"public":[18,19,20,21],"return":[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],"super":[4,20],"target_\u03b8":7,"true":[2,3,4,5,9,10,12,13,18,19,20,21],"try":[1,2,5,18,21],"val_\u03bb2":20,"validation_\u03b6":4,"var":[3,7,20],"while":[4,12,13,17,18,20,21],APS:1,And:[3,5,7,18],Axes:18,Doing:0,For:[1,2,3,4,5,7,8,9,10,11,12,13,14,15,16,18,19,20,21],Not:20,One:[17,20],The:[0,1,2,3,4,5,7,8,9,10,11,12,13,14,15,16,17,18],Then:[3,4,20],There:[1,12,13,18,20,21],These:[1,2,4,7,9,10,11,12,13,14,15,16,18,19,20,21],Theses:1,Used:18,Uses:18,Using:[0,2,18,20],Will:21,With:[4,9,12,14,16,18,19,20,21],__init__:4,_aggregated_imnn:20,_aggregatedimnn:[19,20,21],_bytes_featur:21,_calculate_f_statist:20,_calculate_loss:20,_check_input:20,_check_loop_var:20,_collect_input:[4,19,20],_construct_deriv:[19,20],_construct_gradi:[4,20],_fit:20,_fit_cond:20,_fit_pbar:20,_get_batch_summari:20,_get_f_statist:20,_get_fitting_kei:[19,20],_get_loss:20,_get_paramet:20,_get_regularis:20,_get_regularisation_strength:20,_imnn:[19,20,21],_initialise_histori:20,_initialise_model:20,_initialise_paramet:20,_int64_featur:21,_model_initialis:20,_opt_initialis:20,_run_with_pbar:20,_run_without_pbar:20,_scatter_plot:18,_set_batch_funct:[19,20],_set_data:[19,21],_set_dataset:[19,21],_set_devic:20,_set_histori:20,_set_input:20,_set_prebuilt_dataset:21,_set_shap:[19,20],_setup_plot:20,_setup_progress_bar:20,_shape:3,_slogdet:20,_updat:20,_update_histori:20,_update_loop_var:20,_update_progress_bar:20,abc:[2,3,4,5,7,18],abl:[1,2,12,13,18,20,21],about:[0,1,7,11,16,21],abov:[1,3,4,10,13,15,19,20,21],above_fiduci:1,abs:3,acceler:[9,10,11,12,13,18,21],accept:[3,4,5,7,18],acceptance_ratio:[7,18],acceptance_reach:18,accepted_paramet:18,access:1,accident:[1,3,9,10,12,13,14,15],accomplish:3,accord:5,accuml:20,accumul:[1,20],accur:[19,20],achiev:[1,20],achosen:18,across:1,activ:[4,9,10,11,12,13,14,15,16,20],actual:[3,5,18,21],adam:[3,4,5,9,10,11,12,13,14,15,16,20],add:[4,12,13,18,20,21],added:[3,4,7,18],addit:[0,2,3,19,20],adopt:17,advis:[20,21],after:[3,4,9,10,11,12,13,14,15,16,20],afterward:2,again:[1,5,19,20,21],against:[11,16],aggreg:[0,4,9,10,11,12,13,19,21],aggregatedgradientimnn:[2,8,12,20,21],aggregatedimnn:20,aggregatednumericalgradientimnn:[2,4,8,13,20,21],aggregatedsimulatorimnn:[1,2,8,20],aim:[0,5],albeit:3,algorithm:[0,1,18,20],align:3,all:[1,3,4,5,9,10,12,13,14,15,18,19,20,21],allow:[1,3,5,7,18,20],almost:1,along:[18,19],alpha:[3,5,7,9,10,11,12,13,14,15,16,18,19,20,21],alreadi:[18,19,20],also:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],although:[1,3,5,9,10,11,12,13,18],alwai:[7,19],ambigu:18,american:1,amount:[1,3,4,5,17,18],amplifi:20,amplitud:[3,7],analysi:3,analyt:[3,7,9,10,11,12,13,14,15,16,20,21],ani:[1,7,12,13,18,19,20,21],anoth:[9,10,12,13,14,15],another_fit_kei:16,anywai:[5,18],apart:[4,18,19,21],api:17,appar:5,append_ax:3,appli:20,apply_fun:3,apply_model:1,approach:1,appropri:18,approx:18,approxim:[1,3,4,5,6,19,20,21],approximatebayesiancomput:[0,3,4,5,7],apr:1,arang:3,arbitrarili:20,aren:18,argnum:[1,3,4,5,9,12,14,20,21],argument:[3,7,21],around:[7,20],arrai:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],artefact:18,articl:1,as_numpy_iter:[1,12,13,19,21],aspect:3,assign:20,assum:20,assumpt:18,astyp:5,asymptot:[0,1],attempt:[3,7,18],attribut:[4,7,18,19,20,21],attributedimnn:20,augment:4,author:1,auto:5,autodifferenti:19,automat:[1,5,10,12,13,15,19,20,21],autotun:[2,10,12,13,19,21],aux:[4,18,20],auxillari:20,avail:[1,2,4,17,18],avoid:[1,12,13,18,19,20,21],awai:5,axes:[18,19,20],axes_grid1:3,axhlin:5,axi:[2,3,4,5,7,9,10,12,13,14,15,18,19,20,21],axis_label:[3,7,18],axvlin:[4,5],back:19,backbon:1,backend:17,backpropag:1,badli:5,balanc:20,ball:[5,7,18],bar:[5,9,10,11,12,13,14,15,16,20],base:[0,1,2,18],batch:[1,2,12,13,18,19,20,21],batch_gradi:20,batch_gradients_with_deriv:20,batch_shap:[12,19,21],batch_summari:20,batch_summaries_with_deriv:20,batch_target_data:20,batch_target_kei:20,bayesian:[3,6],bbox_to_anchor:18,becaus:[1,2,3,4,7,9,10,12,13,14,15,18,20,21],becom:[7,17],been:[9,10,11,12,13,14,15,16,17,18,20],befor:[1,3,4,9,10,11,12,13,14,15,16,19,20],begin:[3,7,9,10,11,12,13,14,15,16],being:[1,4,18,19,20,21],below:[4,10,13,15,19,20,21],below_fiduci:1,benefit:20,benjamin:1,best:[1,2,3,5,9,10,11,12,13,14,15,16,18,20,21],best_w:20,beta:[3,7,9,10,11,12,13,14,15,16,19,20,21],better:[5,18],between:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20],bia:5,bias:18,bin:[4,5,18],bit:[1,3,17,21],bitbucket:[1,17],black:[4,5],block:3,blockwis:[3,7,18],bodi:20,boldsymbol:[3,18,20],book:20,bool:[2,18,19,20,21],both:[3,4,5,21],bottom:5,bound:7,box:3,broadcast:[2,3,18,20],bug:[6,18],build:[3,4,9,10,11,12,13,14,15,16,18,19,20,21],built:[3,5,9,11,12,14,16,17,20,21],bulk:5,bunch:4,byte_list:21,c_a:3,c_b:3,cach:[1,2,9,10,12,13,19,21],calcul:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],calculate_loss:20,call:[4,9,10,11,12,13,14,15,16,18,20,21],came:7,can:[0,1,2,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],cannot:[19,20],carefulli:[12,13],carlo:6,caus:[12,13,18],cax:3,cdot:3,centr:18,central:1,certain:7,certainli:1,chain:[1,9,12,14,19,20,21],chanc:20,chang:[4,7,17],channel:3,charnock2018:1,charnock:[1,3],cheap:[1,19,21],check:[3,5,12,13,18,19,20,21],check_func:21,check_param:21,check_siz:21,checkout:18,child:20,choic:[1,18,20],choos:[3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],chosen:[3,16,18],cinv:3,cite:0,clearli:5,clone:17,close:[4,5,9,10,11,12,13,14,15,16,20],closer:18,closest:[5,18],code:[1,3,17],coincid:[18,20],collat:19,collect:[20,21],color:[4,5],colorbar:3,colour:[3,18,20],column:[18,20],com:18,combin:[1,18,19,20,21],come:18,compar:[1,5,18,19,20,21],comparison:[1,3,5,20],compat:17,compil:[1,19,20],complet:[4,17,20],complex:[0,1],complic:21,compress:[1,4,7,18,19,20],compressor:[3,4,5,7,18],comput:[1,3,6,12,19,20,21],concaten:[3,18,19],condit:[1,3,20],confid:18,config:3,conform:20,conj:3,consid:[3,12,13,18,20],consist:5,constant:1,constant_valu:3,constitu:20,constrain:[19,20,21],constraint:[1,3,5,19,20,21],construct:[0,1,2,12,18,19,20,21],contain:[1,2,18,20,21],content:20,continu:[9,10,11,12,13,14,15,16,18,21],contour:[7,18],control:[1,3],conv1x1:3,conv3x3:3,conv5x5:3,conv:3,conveni:8,convent:3,converg:[1,3,4,9,10,11,12,13,14,15,16,18,20],convert:21,convolut:3,cool:5,copi:17,corner:18,correct:[2,3,12,13,18,19,20,21],correctli:[18,19,20],correl:[1,9,10,12,13,14,15],correspond:[4,18,21],cos:3,cosmo:3,cosmo_param:3,cosmolog:[0,2],could:[1,2,3,4,9,10,11,12,13,18,20],count:20,counter:[1,20,21],counterpart:[2,19,21],coupl:[4,9,10,11,12,13,14,15,16,20,21],cours:[1,12,13],covari:[1,2,3,4,9,10,11,12,13,14,15,16,18,19,20,21],cpu:[1,2,3,4,9,10,11,12,13,20,21],cramer:7,creat:[3,18,19,20,21],criterion:[18,20],cuda:17,cumul:18,current:[1,5,6,12,13,18,20,21],current_paramet:1,custom:3,cutidx:3,cutoff:18,d_i:[7,9,10,11,12,13,14,15,16],dark:3,dash:[4,5,7],dashdot:7,data:[0,1,2,3,4,5,7,8,9,10,11,12,13,14,15,16,18,19,20],data_kei:[9,10,12,13,14,15,20,21],dataset:[0,2,4,11,12,13,16,17,19,20],datasetgradientimnn:[0,1,2,8,20],datasetlgradientimnn:12,datasetnumericalgradientimnn:[0,1,2,8,20],dateset:21,deal:17,decai:1,decid:3,decis:5,decor:[18,20],def:[1,3,4,5,7,9,10,11,12,13,14,15,16,20,21],defin:[3,4,7,18,19,20,21],delta:[3,4,10,12,13,15,19,21],delta_:[19,21],demonstr:1,dens:[4,5,9,10,11,12,13,14,15,16,20],densiti:[4,5,18],depend:[2,17,18,20,21],der:[1,4,12,13,21],deriv:[1,2,3,4,7,9,10,11,12,13,14,15,16,19,20,21],derivative_:[1,12,21],derivative_batch_shap:[4,19,21],derivative_einsum:20,derivative_iter:[19,21],derivative_output_shap:[19,21],derivative_pars:[1,12,21],derivative_serialis:21,derivative_shap:[13,21],describ:[2,7,9,10,11,12,13,14,15,16,18,20,21],descript:20,design:[1,3,20],desir:[2,18,19,20,21],det:3,detail:20,detc:20,detect:20,determin:[1,3,4,9,10,11,12,13,14,15,16,18,19,20,21],detf:20,detinvc:20,detriment:20,develop:[1,17],deviat:20,devic:[1,2,4,9,10,11,12,13,19,20,21],devicearrai:[3,7,20],dft:3,diag:[1,3,10,13,15,20,21],diagon:18,dict:[3,20],dictionari:20,differ:[1,2,3,4,5,8,9,10,11,12,13,14,15,16,18,19,20,21],differenti:[3,4],difficult:[1,7,10,13,15],digress:18,dim:3,dimens:[3,20],dimension:[0,1],direct:[3,4,5,18,20,21],directli:[1,12,13,20],directori:[12,13,20,21],direrectli:19,disk:1,displaystyl:[7,9,10,11,12,13,14,15,16],dist:18,distanc:[7,18,20],distance_measur:18,distant:18,distribut:[0,1,3,4,5,7,18,20],divid:[3,19],dlength:3,do_3x3:3,do_5x5:3,docstr:20,document:[17,20],doe:[1,5,12,13,18,20,21],doesn:[1,5,20],doi:1,doing:[1,18],domain:18,don:1,done:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20,21],dot:[3,7],down:[1,3,18,21],draw:[4,7,18,20],drawn:[4,5,18],driven:18,dtype:[3,7,20],dual:[19,20,21],due:[3,4,9,10,11,12,13,14,15,16,17,20,21],duplic:20,dure:[17,20],dx_dw:[4,20],dynam:[19,20,21],each:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],eachdevic:20,earli:[1,3,5,9,10,11,12,13,14,15,16,17,20],eas:18,easi:20,easier:[1,17,21],easiest:2,easili:[1,4,18,20],edg:18,edit:4,effect:[5,10,13,15],effici:[1,3,18],einsum:[4,10,13,15,20,21],either:[1,2,12,13,17,18,19,20,21],element:[4,18,19,20,21],elif:3,els:[3,4],empti:[18,20],enabl:[17,19,21],encapsul:20,end:[2,3,7,9,10,11,12,13,14,15,16,20],enough:[1,18],ensur:[3,5],entir:20,enumer:[3,12,13,21],envis:18,epoch:5,epsilon:[3,4,5,9,10,11,12,13,14,15,16,18],equal:[3,18,19,21],equival:18,error:[5,12,13,18,20],essent:[2,20],estim:[1,3,5,7,9,10,11,12,13,14,15,16,18,19,20,21],etc:[1,4,12,13,17,18,21],euclidean:18,euclidean_dist:18,evalu:[1,3,7,18,20],even:[5,12,18,21],event_shap:18,everi:[4,18],everyth:[1,20],exact:[7,10,12,13,15,20,21],exactli:[1,5,12,13,18,21],examin:[7,9,10,11,12,13,14,15,16],exampl:[0,1,4,5,7,9,10,11,12,13,14,15,16,17,18,20,21],exceed:21,except:20,exception:1,exist:[1,18,20],exp:[3,7,9,10,11,12,13,14,15,16,19,20,21],expand:18,expand_dim:[3,4,5,7,9,10,12,13,14,15,21],expect:[3,4,7,9,10,11,12,13,14,15,16,20],expected_detf:[1,3,9,10,11,12,13,14,15,16,20],expens:[1,4,9,10,11,12,13,14,15,16,18,19,20],experiment:[1,2,3,4,5,9,10,11,12,13,14,15,16,17,20],explicit:18,explicitli:[3,8,17,20],expon:3,extend:4,extent:3,extra:[4,9,10,12,13,14,15,18,21],extract:[0,1,2,3,5,9,10,12,13,14,15,20],extrem:[1,4,18,19,20,21],f_aa:3,f_ab:3,f_ba:3,f_bb:3,f_distanc:18,fact:[9,10,11,12,13,14,15,16,20],fail:[17,21],failur:[12,13,20],fairli:[1,4,19,21],fals:[3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],faninconcat:3,fanout:3,far:[1,3,5,18,20],farm:[19,21],fast:20,feasibl:1,featur:[1,9,10,11,12,13,14,15,16,17],feed:[19,21],few:[7,18,20],fft:3,fft_norm:3,fftn:3,fid:[3,4,5,7,9,10,11,12,13,14,15,16,19,20,21],fiddli:4,fiduci:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],fiducial_:[1,12,13,21],fiducial_batch_shap:[4,19,21],fiducial_iter:[19,21],fiducial_paramet:1,fiducial_serialis:21,fiducial_shap:[13,21],field:[0,2,7,9,10,11,12,13,14,15,16],fig:[3,5,7],figsiz:[3,5,18,20],figur:[5,18,20],file:[4,12,13,21],filenam:[12,13,20,21],filepath:21,fill:[12,13,18,20,21],filter:3,filters1:3,filters2:3,filters3:3,filters3l:3,filters4:3,filters5:3,final_w:20,find:[7,9,10,11,12,13,14,15,16,18],fine:17,finish:[5,20],finit:1,first:[3,4,5,7,17,18,19,20,21],fisher:[0,1,2,3,4,7,9,10,11,12,13,14,15,16,18,19,20,21],fit:[0,1,2,3,4,5,9,10,11,12,13,14,15,16,19,20,21],fit_kei:[11,16,20],fix:[5,9,10,12,13,14,15,18,19,20,21],flat:18,flat_map:[13,21],flatten:[3,19],flexibl:[1,4,12,13],float32:[3,5,7,12,13,20,21],float64:3,flow:[12,13,21],flukish:20,fly:[0,1,2,5,11,16,19,20],focu:[1,19,20,21],follow:[3,20,21],forc:[18,19,20,21],foreground:3,form:[1,3,11,19,20,21],format:[12,13,18,19,21],found:20,fourier:3,fourier_field:3,frac:[3,7,9,10,11,12,13,14,15,16,19,20,21],fraction:18,framework:[1,20],free:[0,1,2,7],freedom:20,from:[0,1,2,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],from_gener:[12,13,21],from_tensor_slic:[12,13,21],full:[18,21],fulli:[3,18,20],func:[4,20,21],functool:[4,5,7,12,13,21],further:5,furthermor:[9,10,11,12,13,18],furthest:18,futur:20,g_mse:5,gaussian:[1,4,7,9,10,11,12,13,14,15,16,19,20,21],gaussianapproxim:[0,3,7],gener:[0,1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],get:[1,3,4,5,18,19,20,21],get_accept:18,get_deriv:[13,19,21],get_estim:[3,4,5,18,20],get_fiduci:[13,21],get_fil:21,get_gradi:[19,20],get_grids:18,get_initial_se:21,get_keys_and_param:20,get_level:18,get_margin:18,get_min_accept:18,get_paramet:1,get_sampl:18,get_se:21,get_serialis:21,get_sims_and_d:[9,12,14,21],get_simul:[12,13,21],get_summari:[4,19,20],git:17,github:18,give:[4,20,21],given:[3,5,7,18,20,21],glob:[1,12,13,21],going:[3,4,5,7,9,10,11,12,13,14,15,16,20,21],good:18,gpu:[1,4,9,10,11,12,13,20,21],grab:[3,4,9,10,12,13,14,15,21],grabber:21,grad:20,gradient:[0,1,2,4,5,9,10,11,12,13,19,21],gradientimnn:[0,2,8,20,21],greater:20,green:5,grid:18,gridpoint:18,gridsiz:[3,4,7,18],guilhem:1,had:17,hand:7,handi:1,handl:[1,2,20],hard:1,has:[3,5,9,10,11,12,13,14,15,16,18,19,20,21],hat:20,have:[1,3,4,6,9,10,11,12,13,14,15,16,18,19,20,21],heavili:17,hei:5,help:[1,4,9,10,11,12,13,14,15,16],helper:[20,21],henc:18,here:[1,3,5,9,10,11,12,13,14,15,16,18,20,21],hermitian:3,hidden:[4,9,10,11,12,13,14,15,16,20],high:[1,3,4,5,7,18,20,21],highest:[9,10,11,12,13,14,15,16,20],highli:1,hist:[4,5,18],histogram:[5,18],histori:[9,10,11,12,13,14,15,16,20],history_plot:20,histtyp:5,hit:20,hold:18,holder:18,horizon:20,horizont:[5,20],host:[1,2,4,9,10,11,12,13,19,20,21],how:[2,3,4,5,7,9,10,11,12,13,14,15,16,19,20,21],howev:[1,5,21],hspace:5,hstack:3,http:[1,17],huge:[17,19,21],hundr:[12,13,21],i_a:[19,20,21],i_b:[19,20,21],ideal:1,ident:[1,4,9,10,11,12,13,14,15,16,18,20],ifftn:3,ifftshift:3,ijk:[10,13,15,21],ikj:4,im1:3,imagin:[4,20],imnn2021:1,imnn:[7,8,9,10,11,12,13,14,15,16,17,18,20,21],impact:1,implement:[1,3,17,18,20],importantli:21,improperli:21,improv:20,imshow:3,incept:3,inceptblock:3,inceptnet:3,inch:18,includ:[9,10,11,12,13,14,15,16,19,20],incorrect:18,incorrectli:[9,10,12,13,14,15],increas:[9,10,11,12,13,14,15,16,20,21],increment:21,ind:20,independ:[1,4,5,7,19,20,21],index:[0,5,18,21],index_upd:[3,5],indic:[1,3,5,18],individu:[18,21],induc:20,infer:[0,1,2,7],infinit:18,infinitessim:18,inform:[0,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],inherit:[18,19,20,21],init_fun:3,initi:[1,4,9,10,11,12,13,14,15,16,18,20,21],initial_param:20,initial_paramet:1,initial_st:20,initial_w:20,initialis:[1,2,4,5,9,10,11,12,13,14,15,16,18,20],initialised_paramet:1,inmn:5,innermost:18,input:[1,2,3,4,5,18,19,20,21],input_shap:[1,2,3,4,5,9,10,11,12,13,14,15,16,19,20,21],insert:[7,9,10,11,12,13,14,15,16,20],insid:[5,18],instal:[0,3,18],instanc:18,instanti:5,instead:[1,7,18,20],int_list:21,integ:[4,20],intend:1,inter:20,interact:2,interest:[3,4,7,9,10,11,12,13,14,15,16,21],introduc:[3,11,16],inv:[3,7,18],invc:[3,20],invers:[7,18,20],invert:20,invf:[3,7,18,20],involv:18,inward:7,isinst:3,isn:[12,13,17,20],isnan:7,isocontour:18,isscalar:3,issn:1,issu:[3,9,10,11,12,13],iter:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],ith:11,its:[1,3,4,11,12,18,19,20,21],itself:[3,20],jac:17,jacobian:[1,19,21],jacrev:4,jax:[1,2,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],jax_cosmo:3,jax_enable_x64:3,jaxlib:17,jit:[5,19,20],joint:[5,18],journal:1,just:[4,5,12,20,21],justifi:[19,20,21],justinals:18,k_a:3,k_b:3,k_shape:3,keep:[12,13,20,21],kei:[1,3,4,5,7,9,10,11,12,13,14,15,16,19,20,21],kept:18,kera:17,kernel:3,key_or_st:[1,2,3,4,5,9,10,11,12,13,14,15,16,19,20,21],know:[3,4,5,7,9,10,11,12,13,14,15,16,20,21],knowledg:[7,18],known:[3,17,18,19,21],kwarg:[3,4],label:[3,5,7,18,19,20,21],lambda:[1,3,5,7,9,10,11,12,13,14,15,16,19,20,21],lambda_2:[19,20,21],langl:3,languag:17,larg:[1,2,3,4,7,9,10,11,12,13,18,19,20,21],larger:[4,7,9,10,11,12,13,14,15,16,20],last:[3,5,7,19,20],lavaux:1,law:2,lax:20,layer:[3,4,9,10,11,12,13,14,15,16,20],lead:5,leaki:[4,9,10,11,12,13,14,15,16,20],leakyrelu:[3,4,5,9,10,11,12,13,14,15,16,20],learn:[1,19,21],least:[1,5,18,21],leav:[9,10,11,12,13,14,15,16],left:[3,7,9,10,11,12,13,14,15,16,18],leftarrow:[4,5,18],legend:[5,18,20],len:[2,3,12,13,20,21],length:[3,4,9,10,11,12,13,14,15,16,18,19,20,21],less:[5,18,21],let:[2,3,4,7,9,10,11,12,13,20],level:18,lfi:[2,3,4,5,7,17,18],librari:3,like:[1,2,3,5,7,9,10,11,12,13,14,15,16,17,18,20,21],likelihood:[0,1,2,3,5,7,9,10,11,12,13,14,15,16,20],likelihoodfreeinfer:[0,3,7],lim_:18,limit:[1,3,4,9,10,11,12,13,14,15,16,17,18,20],linalg:[3,7,18],line:[4,5,18,20],linear:[3,9,10,11,12,13,14,15,16,19,20,21],linear_matter_pow:3,linestyl:[4,5,7,18],linspac:[4,5],list:[1,2,4,12,13,18,19,20,21],littl:21,load:[0,1,12,13,20],loc:5,log:[0,1,2,19,20,21],log_likelihood:3,log_norm:3,log_prob:7,logarithm:[19,20,21],lognorm:3,look:[1,5,9,10,12,13,14,15],loop:[1,12,18,19,20,21],loopabl:19,loss:[1,5,19,20,21],lossi:[9,10,11,12,13,14,15,16],lossless:1,losslessli:0,lot:[1,18],low:[3,4,5,7,18,20],lower:7,luca:3,m_abc:4,m_fid:[4,5],m_kei:5,machin:[19,21],made:[1,4,5,10,12,13,15,18,19,20,21],mag:3,magnitud:1,mai:[1,11,16,17,18,20],main:[1,2,12,17,19,20,21],make:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],make_axes_locat:3,makinen:3,mani:[1,2,4,5,9,10,11,12,13,14,15,16,18,19,20,21],manipul:18,manual:[5,20],map:[0,1,12,13,20,21],margin:[5,7,18],marginal_plot:[3,7,18],marginalis:18,marker:18,massiv:[9,10,11,12,13,14,15,16,18,20],match:[1,10,13,15,19,20],mathcal:[3,7,9,10,11,12,13,14,15,16],matplotlib:[3,4,5,7,18],matric:18,matrix:[1,3,4,9,10,11,12,13,14,15,16,18,19,20,21],matter:3,max:[5,18],max_accept:18,max_detf:20,max_iter:[3,5,7,9,10,11,12,13,14,15,16,18,20],max_sampl:18,maxim:1,maximis:[0,19,20,21],maximum:[1,3,9,10,11,12,13,14,15,16,18,20,21],maxmimum:20,maxpool:3,maxval:[3,4,20],mayb:[1,18],mean:[1,2,4,5,6,9,10,11,12,13,14,15,16,18,19,20,21],mean_kei:20,mean_val:20,meaning:18,meant:[1,20],measur:18,memori:[1,2,4,9,10,11,12,13,19,20,21],meshgrid:3,method:[1,9,10,11,12,13,14,15,16,18,19,20,21],might:[1,3,7,20],min:5,min_accept:[3,5,7,18],min_iter:[3,9,10,11,12,13,14,15,16,20],minimum:[7,18,20],minval:[3,4,20],mislead:[3,18],mode:3,model:[0,2,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],model_kei:[9,10,11,12,13,14,15,16,20],model_simul:20,modul:[0,1,3,4,7,9,10,11,12,13,14,15,16,17],momentum:1,mont:6,month:1,mope:[9,10,11,12,13,14,15,16],more:[1,3,5,7,10,13,15,17,18,20,21],most:[0,1,18,20],mostli:18,move:[7,18],move_sampl:18,mpl:20,mpl_toolkit:3,mse:5,mse_abc:5,mse_loss:5,mse_m:5,mse_output:5,mse_stat:5,mse_w:5,mu_a:[19,20,21],mu_b:[19,20,21],mu_i:20,much:[1,3,10,13,15],multipl:[1,18,20],multiple_el:20,multipli:[18,20],multius:20,multivari:18,must:[1,3,4,7,12,13,18,19,20,21],mx_mp:4,n_column:18,n_d:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,19,20,21],n_devic:[1,4,12,13,19,20,21],n_i:3,n_initial_point:[7,18],n_iter:[19,20,21],n_nois:4,n_parallel_simul:18,n_param:[1,2,3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],n_per_devic:[1,2,4,9,10,11,12,13,19,20,21],n_point:[7,18],n_remain:[19,21],n_remaining_iter:[19,21],n_s:[1,2,3,4,5,9,10,11,12,13,14,15,16,19,20,21],n_sampl:[3,5,7,18],n_scale:3,n_sim:[12,13,21],n_simul:18,n_summari:[1,2,3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],n_target:[3,7,18],nabla_:3,name:[18,20,21],natur:[17,21],ncol:[3,18,20],nearli:3,necessari:[9,10,12,13,14,15,17,19,20,21],necessarili:[1,18],need:[1,3,4,5,7,10,11,12,13,14,15,16,18,19,20,21],neg:[7,9,10,11,12,13,14,15,16,19,20,21],network:[0,2,3,4,9,10,11,12,13,14,15,16,18,19,20,21],networki:1,neural:[0,2,3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],neuron:[4,9,10,11,12,13,14,15,16],never:1,newaxi:[3,5],newer:17,next:[4,18,21],nice:3,nois:[0,2,3,20,21],noiseimnn:4,noisenumericalgradientimnn:20,noisi:[4,5],non:[7,9,10,11,12,13,14,15,16,18],none:[1,2,3,4,7,9,12,13,18,19,20,21],norm:3,normal:[0,2,4,5,7,9,10,11,12,13,14,15,16,18,20,21],normalis:[3,18],note:[0,1,3,5,7,9,10,12,13,14,15,16,20,21],notic:3,now:[3,4,5,7,9,10,11,12,13,14,15,16,21],npy:[12,13,21],nrm:3,nrow:3,num:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20,21],num_parallel_read:[1,12,13,21],number:[0,1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],numer:[1,2,3,4,9,10,11,12,13,15,19,20,21],numerical_deriv:[10,13,15,21],numerical_derivative_:[13,21],numerical_derivative_pars:21,numericalgradientimnn:[0,2,8,20,21],numpi:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,19,20,21],obj:[2,20],object:[1,2,18,19,20,21],obs:18,observ:[3,7,18],obtain:[1,3,4,7,9,10,11,12,13,14,15,16,18,19,20,21],occur:20,off:[5,18,19,20,21],ofnetwork:2,often:20,omega_c:3,omega_m:3,onc:[1,2,7,9,10,12,13,14,15,18,19,20,21],one:[2,3,5,9,10,11,12,13,14,15,16,18,20],ones:[3,18],onli:[1,3,9,10,11,12,13,18,19,20,21],onto:[19,20],open:[12,13,21],oper:[1,9,10,11,12,13,20],ops:[3,5],opt:21,optim:[1,2,3,4,5,9,10,11,12,13,14,15,16,18,20],optimis:[1,2,3,4,5,9,10,11,12,13,14,15,16,19,20,21],option:[2,18,20],order:[1,4,7],org:[1,17],orient:[3,5],origin:[18,19,21],other:[1,12,17,18,19,20,21],otherwis:[16,18,20],ouput:[19,20],our:[3,4,5,7,9,10,11,12,13,14,15,16,21],out:[3,4,9,10,11,12,13,14,15,16,18,19,20,21],outer:[3,18],outlin:[19,20,21],output:[1,2,3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],output_shap:[1,20],outsid:18,over:[1,3,9,10,11,12,13,18,19,20,21],overcom:20,overestim:[1,18],overfit:[11,16],overflow:[2,3,20],overli:20,overrid:20,overwritten:[18,21],own:[18,21],pad:[3,21],page:[0,2],pair:[1,3,10,13,15,18],panel:20,parallel:[3,18],parallelis:18,param:[0,1,3,13,18,19,20,21],paramet:[0,1,2,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],parameter_differ:1,parameter_estim:[3,7,18],parameter_kei:[5,7],parameteris:1,parent:[0,4],pars:[1,21],parser:[1,12,13,21],part:[3,12,18,21],partial:[3,4,5,7,12,13,18,19,20,21],particular:[0,7,12,18,19,20],particularli:5,pass:[1,2,3,4,5,12,13,18,19,20,21],path:21,patienc:[9,10,11,12,13,14,15,16,20],patience_count:20,pbar:20,per:[1,4,9,10,11,12,13,18,20],percentag:18,percentil:[7,18],perform:[1,2,3,12,13,18,20],perturb:[2,19,21],pha:3,phi:3,phi_:3,physic:[1,2,9,10,12,13,14,15,18,19,20,21],physrevd:1,piec:5,pip:[3,17,18],pix:3,pixel:3,place:[18,19,20],planck15:3,pleas:[1,21],plenti:17,plot:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20],plotter:18,plt:[3,4,5,7],pmc:[7,18],png:20,point:[1,5,7,9,10,11,12,13,14,15,16,18,20],popul:6,populationmontecarlo:[0,7],posit:18,possibl:[1,2,4,5,18,20,21],posterior:[4,5,7,18],power:[0,2],pre:18,preassign:21,prebuilt:20,precalcul:[1,18,20],precis:[1,21],preconstruct:21,pred:5,predefin:[19,20],prefetch:[1,2,9,10,12,13,19,21],preinitialis:[2,20],premad:[4,18],prepend:21,prerun:18,preset:18,pretti:1,prevent:[1,7,11,16,18,20,21],previou:[17,18,20],principl:5,print:[3,5,7,18,20,21],print_rat:[3,4,5,9,10,11,12,13,14,15,16,20],prior:[3,4,5,7,18],privat:[18,19,20,21],prngkei:[1,3,4,5,7,9,10,11,12,13,14,15,16,20,21],probabl:[4,5,7,17,18,19,20],problem:[1,7,9,10,11,12,13,14,15,16,17,18,20],procedur:[19,20,21],process:[1,4,8,9,10,11,12,13,14,15,19,21],prod:3,prod_i:[7,9,10,11,12,13,14,15,16],produc:[0,20,21],product:3,progress:[9,10,11,12,13,14,15,16,20],proper:[20,21],properli:4,propog:18,propos:[7,18],propto:7,provid:[1,2,3,4,5,18,19,20,21],pseudo:20,publish:1,pull:[3,17],purpos:2,push:[19,20],put:[1,20],put_margin:[3,7,18],pydelfi:18,pyplot:[3,4,5,7],pytest:18,python3:17,python:17,pytre:1,quasi:[1,20],quick:[19,20,21],quicker:[1,12,13,21],quickest:21,quickli:[1,20],quit:[1,4,12,13,17,21],r_p:3,rais:[18,19,20,21],randint:[3,4],random:[1,2,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],randomli:[4,5],rang:[1,3,4,5,7,12,13,18,20,21],rangl:3,rao:7,rate:[1,7,20],rather:[3,5,18,20,21],ratio:18,ravel:3,raveled_rang:7,reach:[3,7,12,13,18,20,21],read:[12,13,21],readi:18,real:[3,5],realis:[1,3,4,5,10,13,15,20],realist:3,realiti:3,realli:[1,5,9,10,11,12,13,21],reason:[1,3,5,20,21],recollect:[19,21],recommend:[1,20],record:[12,13,21],record_s:[12,13,21],red:5,reduc:[9,10,11,12,13,14,15,16,18,20],reflect:18,region:18,regress:[0,2],regularis:[1,3,4,9,10,11,12,13,14,15,16,19,20,21],reinterpreted_batch_ndim:[4,5,7],reject:[7,18],rel:[1,20],relat:[7,9,10,11,12,13,14,15,16],relu:[4,9,10,11,12,13,14,15,16,20],rem:12,remain:[1,2,12,18,19,21],remaind:20,remaining_:12,remaining_batch_shap:[12,19,21],remov:[3,18],repeat:[1,4,7,9,10,12,13,14,15,21],replac:[18,20],repositori:17,repres:[18,20],represent:21,requir:[1,2,4,17,18,20,21],rerun:[9,10,11,12,13,14,15,16],rescal:[3,18,19,20,21],reset:[4,20],reshap:[3,4,7,10,12,13,15,19,20,21],resort:[10,13,15],respect:[1,2,3,4,9,10,11,12,13,14,15,16,19,20,21],restart:20,result:[1,5,9,10,11,12,13,14,15,16,18,20,21],ret_check:20,retrain:1,retriev:18,revidx:3,review:1,rgb:20,right:[3,5,7,9,10,11,12,13,14,15,16],rightarrow:3,rng:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],root:[3,20],routin:[4,10,11,12,13,14,15,16,20],row:18,rule:[1,9,12,14,19,20,21],run:[1,3,4,5,9,10,11,12,13,14,15,16,18,20,21],run_with_pbar:20,run_without_pbar:20,safest:21,sai:[3,4,7,9,10,11,12,13],same:[1,3,4,5,10,13,15,18,19,21],sampl:[5,7,18],sampler:18,save:[4,12,13,17,20,21],sbi:18,scalar_output:20,scale:[3,4,9,10,11,12,13,14,15,16,18,19,20,21],scan:20,scatter:[5,18],scatter_plot:[7,18],scatter_summari:[7,18],scenario:[1,21],scheme:18,score:[3,20],search:0,second:[3,7,9,10,11,12,13,14,15,16,20],see:[2,3,5,20,21],seed:[1,5,7,10,12,13,15,16,19,20,21],seem:[6,18,20],seemingli:5,seen:[1,5,7],select:[2,19],self:[4,20],semilog:5,sens:[1,18],sensibl:18,sensit:1,separ:[1,4,10,13,15,20,21],sequenti:21,serial:[3,4,5,9,10,11,12,13,14,15,16,20],serialis:21,serv:[2,19,21],set:[0,1,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],set_accept:18,set_f_statist:20,set_margin:18,set_postfix:5,set_sampl:18,set_text:3,set_xtick:3,set_ytick:3,setup:[1,17,20],setup_plot:18,sever:[1,8,9,10,11,12,13,18,21],shape:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],shortli:1,should:[1,2,3,4,5,9,10,11,12,13,14,15,16,17,18,19,20,21],shouldn:18,show:[3,4,5],shuffl:1,shut:21,side:[1,3],sigma8:3,sigma:[7,9,10,11,12,13,14,15,16,20],sigma_8:3,sigmoid:20,sign:20,signal:4,signific:[18,20],significantli:18,sim:[1,7,9,10,11,12,13,14,15,16,18],similar:3,similarli:[7,18],simpl:[4,9,10,11,12,13,14,15,16,18],simplest:21,simpli:[3,4,9,10,11,12,13,14,15,16,18,20,21],simplic:7,simul:[1,2,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],simulation_gradi:3,simulator_arg:[3,20],simulator_kei:[5,7],simulatorimnn:[0,2,3,5,8,11,20],simultor:3,sin:3,sinc:[1,3,5,7,9,10,11,12,13,14,15,16,18,19,20,21],singl:[1,2,3,4,9,10,11,12,13,18,19,20,21],single_el:20,single_iter:18,single_iteration_condit:18,situat:1,size:[1,2,3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],skip:[1,21],slice:[3,4,5],slightli:[1,3],slope:[4,5],slow:[12,13,21],small:[1,7,18,19,20,21],smaller:5,smooth:[3,7,18,19,20,21],societi:1,softmax:3,softwar:1,solid:18,some:[0,1,3,4,5,7,9,10,12,13,14,15,17,18,20,21],someth:[1,2,20,21],somewhat:18,sort:[1,3,12,13,18,21],space:[0,3,5,18],specif:[0,1,2,4,5,12,13,18,20],specifi:[18,20],spectra:3,spectrum:[0,2],speed:[1,2],split:[1,3,4,5,7,9,10,11,12,13,14,15,16,20,21],spuriou:[1,11,16],sqrt:[3,4,7,9,10,11,12,13,14,15,16,20,21],squar:[3,5,9,10,11,12,13,14,15,16,18,20],squeez:[3,5],stabil:20,stabl:[1,7,17],stack:[1,3,4,5,7,20],stand:20,standard:20,start:[7,20,21],stat:18,state:[1,2,20],stateless:[1,2,20],static_argnum:20,stationari:18,statist:[1,7,9,10,11,12,13,14,15,16,19,20,21],stax:[1,2,3,4,5,9,10,11,12,13,14,15,16,20],steep:20,step:[1,3,4,5,9,10,11,12,13,14,15,16,18,20,21],step_siz:[3,4,5,9,10,11,12,13,14,15,16,20],still:[1,17],stochast:[3,19,20],stop:[1,3,5,9,10,11,12,13,14,15,16,18,20],store:20,str:[18,20,21],strength:[1,4,9,10,11,12,13,14,15,16,19,20,21],stride:3,string:21,stuff:1,subclass:[0,1,2,3,20],sublot:20,submodul:[2,7],subplot:[3,5,7,18,20],subplots_adjust:5,substrat:[3,4,5,7,18],subtract:19,suffici:[7,9,10,11,12,13,14,15,16,19,20,21],suggest:18,suit:17,sum:[1,3,4,9,10,11,12,13,14,15,16,18,20],sum_:[3,19,20,21],sum_i:[7,9,10,11,12,13,14,15,16],summari:[0,2,3,4,5,9,10,11,12,13,14,15,16,18,19,20,21],summaris:[2,3,4,5,6,19,20,21],suppli:18,sure:[3,9,10,12,13,14,15,18],swap:19,sys:[3,21],tag:21,take:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,21],taken:[20,21],tanh:20,target:[0,1,3,4,5,7,18,20],target_choic:18,target_data:[3,4,5,7,18,20],target_kei:20,target_m:[4,5],target_summari:[5,18],tend:18,tensor:21,tensorflow:[2,4,7,10,12,13,17,18,19,20,21],tensorflow_prob:[3,4,5,7,18],term:[1,9,10,11,12,13,14,15,16,20],test:[5,18],test_i:5,test_m:5,textrm:[3,4,5,7,9,10,11,12,13,14,15,16,18],tfd:7,tfp:[3,4,5,7,18],tfrecord:[0,1,12,13],tfrecorddataset:[1,12,13,21],than:[1,3,4,5,9,10,11,12,13,14,15,16,18,20,21],thei:[1,4,5,7,9,10,11,12,13,14,15,16,17,18,21],them:[1,3,4,18,19,20,21],therefor:[1,4,7,9,10,11,12,13,19,20],theta:[3,18,19,20,21],theta_:[3,19,20,21],thi:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],thing:[1,3,5,20],those:[18,19],though:21,three:[4,5,9,10,11,12,13,14,15,16,20],through:[1,4,5,12,13,19,20,21],tick_param:5,tight:1,tighter:5,tild:3,tile:[7,10,13,15,21],time:[1,2,3,4,5,9,10,11,12,13,14,15,16,17,18,19,20,21],times128:3,titl:[1,3],tmp:[12,13,21],to0:[3,18],togeth:[1,19,20],tom:[1,3],tomcharnock:[1,17],too:[4,9,10,11,12,13,18,19,20,21],took:5,tool:[0,2],top:5,total:[3,4,7,12,13,18,20,21],total_draw:18,toward:18,tpl:3,tqdm:[5,20],trace:3,tracer:20,train:[0,1,2,4,5,9,10,11,12,13,14,15,16,18,20],trang:5,transform:[1,3,4,19,20],transpar:18,tri:18,tricki:1,trivial:18,truncat:[18,20],tupl:[1,2,3,18,19,20,21],turn:[9,10,11,12,13,14,15,16,18,19,20,21],two:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20,21],type:[1,2,3,18,19,20,21],typeerror:[18,19,20,21],unbias:1,uncertainti:20,uncoupl:3,under:[1,18,20],underestim:18,undersampl:18,unfortun:20,uniform:[3,4,5,7,18,20],unimport:20,uniqu:3,unit:21,univers:3,unknown:[7,9,10,11,12,13,14,15,16],unless:18,unlik:18,until:[1,3,5,12,13,18,21],unus:19,unusu:20,unwrap:[20,21],updat:[1,2,3,18,19,20],updated_st:1,upper:[7,20],upto:20,url:1,use:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],used:[1,3,4,5,9,10,11,12,13,18,19,20,21],useful:[2,3],user:[3,17],uses:[1,7,17,20],using:[1,3,4,5,7,8,9,10,11,12,13,14,15,16,18,19,20,21],util:[1,3,9,12,14,17,18,21],val_der:12,val_detc:20,val_detf:20,val_detinvc:20,val_fid:12,val_r:20,val_rem:12,valid:[1,2,4,5,9,10,12,13,14,15,19,20,21],validation_:21,validation_deriv:[1,2,4,9,10,12,13,14,15,19,21],validation_derivative_:[12,21],validation_fiduci:[1,2,4,9,10,12,13,14,15,19,21],validation_fiducial_:[12,13,21],validation_i:5,validation_kei:4,validation_loss:20,validation_m:5,validation_main:[1,2,12,19,21],validation_mx:4,validation_mx_mp:4,validation_numerical_deriv:[10,13,15,21],validation_numerical_derivative_:[13,21],validation_remain:[1,2,12,19,21],validation_remaining_:12,validation_simul:[13,21],valu:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],value_and_grad:5,value_and_jac:17,value_and_jacfwd:[3,9,12,14,21],value_and_jacrev:[1,3],valueerror:[18,19,20,21],var_kei:20,var_val:20,vari:[1,10,12,13,15,19,21],variabl:[4,11,16,20],varianc:[2,4,6,9,10,11,12,13,14,15,16,18,20,21],variant:1,variat:[10,13,15],varied_abov:1,varied_below:1,varieti:3,variou:[20,21],varphi:3,varphi_:3,varphi_k:3,vector:[4,9,10,11,12,13,14,15,16,19,21],vectoris:20,verbos:[2,18],veri:[1,3,4,5,7,9,10,11,12,13,14,15,16,18,20],version:[1,17,18,20],vertic:[3,5,20],via:[1,2,3,4,16,17,19,20,21],visual:3,visualis:[9,10,11,12,13,14,15,16],vmap:[1,3,5,7,9,10,12,13,14,15,20,21],vol_norm:3,volum:[1,3],wai:[1,2,4,5,8,10,11,12,13,15,16,18,20,21],wall:3,wandelt:1,want:[1,3,4,7,12,13,17,19,20,21],warn:20,weight:[1,18,20],well:[1,3,4,5,7,9,10,11,12,13,14,15,16,17,19,20],what:[3,7,9,10,11,12,13,14,15,16,20],when:[1,3,10,13,15,18,19,20,21],where:[0,1,2,3,4,5,7,9,10,11,12,13,14,15,16,18,19,20,21],whether:[1,2,3,18,19,20,21],which:[1,2,3,4,5,7,9,10,11,12,13,14,15,16,17,18,19,20,21],whichev:17,while_loop:20,whilst:[2,5,7,19,20,21],white:3,whole:[1,3,5,7,18,19,21],wholli:18,whose:[4,5,9,10,11,12,13,14,15,16,18],wide:5,within:[5,7,10,13,15,18,20],without:[2,10,13,15,17,18,19,20,21],won:[3,19],work:[0,2,5,7,17,18,20],wors:7,worst:21,would:[4,5,9,10,11,12,13,14,15,16],wrapper:[8,18,20],write:[12,13,18,21],write_record:[12,13,21],writer:[0,1,12,13],written:[1,3,20,21],wrong:[18,19,20,21],wrt:[19,20,21],wspace:5,x_c:3,x_d:3,x_i:20,x_j:20,x_mp:19,xla:[1,4,19,20,21],xlabel:[4,5],xlim:[5,7],xtick:[5,7],y_abc:4,y_kei:5,y_target:4,year:1,yet:[3,17],yield:[4,12,13,21],ylabel:[5,7],ylim:5,you:[1,17,21],your:21,ytick:5,zero:[3,4,5,12,18,19,20,21],zeros_lik:7,zeroth:[2,18],zeta:4,zip:[1,3,4,12,13,18,21],zorder:5},titles:["Welcome to IMNN\u2019s documentation!","IMNN: Information maximising neural networks","Examples","Inferring cosmology power spectrum parameters from 2D fields","Constructing an IMNN with on-the-fly additive noise","IMNN vs regression networks","Using the likelihood free inference tools","Summarising the mean and the variance","Using specific subclasses","Using the AggregatedGradientIMNN","Using the AggregatedNumericalGradientIMNN","Using the AggregatedSimulatorIMNN","Using the DatasetGradientIMNN","Using the DatasetNumericalGradientIMNN","Using the GradientIMNN","Using the NumericalGradientIMNN","Using the SimulatorIMNN","Installing","Likelihood-free inference modules","Specific IMNN classes","Parent classes","Using tf.data.Datasets for loading"],titleterms:{"class":[19,20],"new":4,Doing:18,The:[19,20,21],Using:[1,6,8,9,10,11,12,13,14,15,16,21],addit:4,aggreg:[1,20],aggregatedgradientimnn:[9,19],aggregatednumericalgradientimnn:[10,19],aggregatedsimulatorimnn:[11,19],approxim:[7,18],approximatebayesiancomput:18,avail:[19,20,21],base:20,bayesian:[7,18],being:2,bug:7,carlo:[7,18],cite:1,comput:[7,18],construct:4,content:0,cosmolog:3,current:7,data:21,dataset:[1,21],datasetgradientimnn:[12,21],datasetnumericalgradientimnn:[13,21],document:0,exampl:[2,3,6,8],few:2,field:3,fly:4,free:[6,18],from:3,gaussian:[3,18],gaussianapproxim:18,gradient:20,gradientimnn:[1,14,19],have:7,here:2,imnn:[0,1,2,3,4,5,19],indic:0,infer:[3,4,5,6,18],inform:1,initialis:3,instal:17,jax:3,law:3,likelihood:[6,18],likelihoodfreeinfer:18,load:21,log:3,maximis:1,mean:7,model:1,modul:[18,19,20,21],mont:[7,18],network:[1,5],neural:1,nois:4,normal:3,note:[17,18],numericalgradientimnn:[1,15,19],paramet:3,parent:20,popul:[7,18],populationmontecarlo:18,power:3,random:3,regress:5,seem:7,simul:3,simulatorimnn:[1,16,19],specif:[8,19],spectrum:3,subclass:8,summaris:7,tabl:0,tensorflow:1,tfrecord:21,todo:[18,20,21],tool:6,train:3,used:2,varianc:7,welcom:0,work:1,writer:21}})