distrax/__init__.py,sha256=9uwpzMSf8O9JOIkl5WFxDqpejGgid_w8SJmfwjicDx4,6773
distrax/distrax_test.py,sha256=8ieYW9HmKJAXKqpGRIkuSz8iXwcKpHv2RyUWd_038TE,982
distrax/_src/__init__.py,sha256=IV2YQAXXaG5v_BQRi14erjv-eiLmBA0Vauu42sj2yIA,696
distrax/_src/bijectors/__init__.py,sha256=IV2YQAXXaG5v_BQRi14erjv-eiLmBA0Vauu42sj2yIA,696
distrax/_src/bijectors/bijector.py,sha256=ulj2bYccDxqbgZFQDRg5ql01nunjv6mkpLhZOBrPIe4,7962
distrax/_src/bijectors/bijector_from_tfp.py,sha256=AplyJGaAm4GCv1Td-tprn_Ou6eyXWCMgSTX2oHnWS_s,4406
distrax/_src/bijectors/bijector_from_tfp_test.py,sha256=z5wpBUsJcTFo9vcNDcwaWFyYOoNefP9jxtd3lRh4F4Q,9688
distrax/_src/bijectors/bijector_test.py,sha256=s8vNpFrovwtmO5V_aI23Le-kTsQwap-36egvO2sHW2s,2179
distrax/_src/bijectors/block.py,sha256=Hug3Z_KSkluUKiXle9KvT7WsjpZRjUrX2fg3sKqCw-w,5101
distrax/_src/bijectors/block_test.py,sha256=W84Qt7jS24Y5vVWFFK7b01F0vXnCp9cGXTmgQWBNa0g,5577
distrax/_src/bijectors/chain.py,sha256=4dT14MOfQDE5Y_pLxEQr4dp1bgK_63SYoAn0MJqttTw,4775
distrax/_src/bijectors/chain_test.py,sha256=VSOSgNq-UtuY2eg5cCo3ZysStO-6R-bkW52Xi5SSdC0,10931
distrax/_src/bijectors/diag_linear.py,sha256=xX5_M5PvBCn3FXE2zVe0BUvv4m0o1yshCOfWzshB6tY,3145
distrax/_src/bijectors/diag_linear_test.py,sha256=atk9ATBXRNDPx3yju0Sj36VLkVuGG0k_aHcPCRAO7w4,7510
distrax/_src/bijectors/diag_plus_low_rank_linear.py,sha256=27QRCFnHrzO8doH9pchXs8kJi9vbi4aCUbwoaDHUEQk,9422
distrax/_src/bijectors/diag_plus_low_rank_linear_test.py,sha256=468zErmFXJWTj0Qco83jZ96hHn7_OvIMie-mzd-U3_E,10657
distrax/_src/bijectors/gumbel_cdf.py,sha256=qEGETtZAbBn7HGiTrZv89kROSt5Cq3kmCBX4VsJk-DQ,1896
distrax/_src/bijectors/gumbel_cdf_test.py,sha256=1UFfqR4wG2MMYkOkbPG6Bte4i5qcw_SNBem4WB-G1zQ,5809
distrax/_src/bijectors/inverse.py,sha256=wU0rC4HS8hI1Ei-4nFKBClM9yMW2rkfxvEcyWwkwjlM,3240
distrax/_src/bijectors/inverse_test.py,sha256=cncQQELfEz0LQKKQhVaPUAU6GW0aji-vklDZfUGpT94,9233
distrax/_src/bijectors/lambda_bijector.py,sha256=lPEBv_6Z3wiK9CIVx3SjpcUa8BzxMxYyM04pC7XMNy8,6504
distrax/_src/bijectors/lambda_bijector_test.py,sha256=2v52Rx9fzq-fFi8jOTXgtLS4jpbN4CZTY1almkKUc7g,12944
distrax/_src/bijectors/linear.py,sha256=WROR5DvUiLzQRlvJ_B_C4yWXoako8N5t1YuEEJ5-7E0,2390
distrax/_src/bijectors/linear_test.py,sha256=phMc1dChFQKNKYEAiBBgXjzwUfOOFgV5WAFeanWiGzA,1794
distrax/_src/bijectors/lower_upper_triangular_affine.py,sha256=xzJkmLO1E-Yjo6dmod-AjeksuntYbCOaWakN5ohGmDo,3980
distrax/_src/bijectors/lower_upper_triangular_affine_test.py,sha256=JYlEMkCjHAPwZCXSJ9BJNEbtRSvJmnnXHXgQcpwpzzE,8764
distrax/_src/bijectors/masked_coupling.py,sha256=RwsqhknfYRc5ikEG_WfvGESUx-DzYnZ-71Fdnuwb8oo,7238
distrax/_src/bijectors/masked_coupling_test.py,sha256=bghSKM4kAtpyziVxHvU0ru_Kb7e6k-UAkHvJXOLuTKY,10755
distrax/_src/bijectors/rational_quadratic_spline.py,sha256=MO4PQQ1KJY4OAqnB4a0oMfE078ORytbYOSG-fRNfcag,20159
distrax/_src/bijectors/rational_quadratic_spline_float64_test.py,sha256=p72sG1mNTuNKmlGGwgw7PuMHA9Q-eT9eZFUs7_Dl-c8,2526
distrax/_src/bijectors/rational_quadratic_spline_test.py,sha256=2zmO0oQWv_SNt7_J4sQYToo4t9XhGAFMVOOF3UZ5qrc,13180
distrax/_src/bijectors/scalar_affine.py,sha256=55P9zjEsaTUfFUj29KaEOv2CsEMwp_NWNxeUq3-9X9o,5195
distrax/_src/bijectors/scalar_affine_test.py,sha256=ujm4Oalfv9pYfc5mQD_ZbSvXGk15OxMADlssR1X7qq4,6928
distrax/_src/bijectors/shift.py,sha256=8rTw6DpkK2SwakWhXQbQaXZVOxlUDTiVDKoBqsoAYaw,2810
distrax/_src/bijectors/shift_test.py,sha256=Dye7657LyJ6CBuXuWdTuy9x7mEIYzKYeamPPtbFgnpc,3950
distrax/_src/bijectors/sigmoid.py,sha256=40Qz-exy62M-QbWuyDHBDBQEvHYyRzTZK_bPg6Gf0ZA,3696
distrax/_src/bijectors/sigmoid_test.py,sha256=ArhLKrW3GjnTBzt1Y8x_bzq5KbjK-c38TUeBZHqvusw,5701
distrax/_src/bijectors/split_coupling.py,sha256=VvFZjnEj9Xjn6HBmoQrphZXxR5sZcuHmJDl4w7qVH08,7600
distrax/_src/bijectors/split_coupling_test.py,sha256=0PqSe-CLcwXkHvQ3YnMbUguWpFJUg4XbpQ1wkOFNHiQ,10197
distrax/_src/bijectors/tanh.py,sha256=hogXlEdcynaKuUGtDtmQ3ZDFbr8tB-5d-9Q1xeWYkK4,2915
distrax/_src/bijectors/tanh_test.py,sha256=9RExX_OPXJA6n0_7L88n05YmH61cDZ9dx5-8KSpWW7Y,6209
distrax/_src/bijectors/tfp_compatible_bijector.py,sha256=aTjCON0m_FVP8FwEoWbNF86AxUvEVDmFHpU5h3P8ORw,9133
distrax/_src/bijectors/tfp_compatible_bijector_test.py,sha256=nqFzL80Lwp_pUB17uc_MPqChGw_WK0qGdu5YzuCY6MQ,14341
distrax/_src/bijectors/triangular_linear.py,sha256=64URn-uzqe4K3-hMNBskBvVqjAFvcizpkU_Sv93Uj4s,4978
distrax/_src/bijectors/triangular_linear_test.py,sha256=jIV4auc5HkNer54QPdyH_lNi-h808F-1oOx38cozCSs,8090
distrax/_src/bijectors/unconstrained_affine.py,sha256=QXIhmBQJvWhecuX2ccMBrHK_Kgt50WkRhqgyYqslq14,5035
distrax/_src/bijectors/unconstrained_affine_test.py,sha256=vDZrrAH04YcqNkI_oUualw3UysNRGfy9M7dlfAXh_ws,8199
distrax/_src/distributions/__init__.py,sha256=IV2YQAXXaG5v_BQRi14erjv-eiLmBA0Vauu42sj2yIA,696
distrax/_src/distributions/bernoulli.py,sha256=tGPidu9HVyUIQehGyD6u7MVAPIGqsODpBNOX3aFo1Sk,7641
distrax/_src/distributions/bernoulli_test.py,sha256=SfPU9DXO7xTqZCKWJc9bzhaOKIVyRJCuma0YmJoOdDo,12888
distrax/_src/distributions/beta.py,sha256=PG9GQsjnkN8IU3IrH2q-Vdqf2rt8Qw2SEOjAbOMgz_w,6577
distrax/_src/distributions/beta_test.py,sha256=is8DldgAWgiTa05rYu1VjvULh0uYxgbAxMKYo3W4Qms,10372
distrax/_src/distributions/categorical.py,sha256=CXTYWbHUNFVj0CSQ6yM6f2V_4Lz66WfcrftSqfDhhe8,7932
distrax/_src/distributions/categorical_test.py,sha256=8pfCgnUSnku0x7jKBRgg4iEtzpHTBMexmdo53xIRBF8,22304
distrax/_src/distributions/categorical_uniform.py,sha256=zjzCpHzrsoI0WGce29bxvEHU_IH8SNbqNkIN_cyGu30,5844
distrax/_src/distributions/categorical_uniform_test.py,sha256=jUgmt5wip9QSIeyd7-IslygPBtp8AcDD8wDFQyRBQJY,13040
distrax/_src/distributions/clipped.py,sha256=F1fPaPB549YxJpe5Ye46zX7-mldUu7JT7bPdnO8w9Gk,5316
distrax/_src/distributions/clipped_test.py,sha256=E9TtRzlVZ6pC1qkRiTA-F2XbYcsKE5DDSC5mAxAS7Zo,7221
distrax/_src/distributions/deterministic.py,sha256=c8kVt6U6bt2yHh5Tgk8yQb6C_WYFsxKJIiugc-WgvxY,6189
distrax/_src/distributions/deterministic_test.py,sha256=mGBeaK-NRNSi2jfAWFHeVBThzfOjfo94a3M9aOEh9uc,9858
distrax/_src/distributions/dirichlet.py,sha256=Ftazn6tnfDelZr_oWpkFVlwiu-YmHvPQ2xvdBS54Z8U,8014
distrax/_src/distributions/dirichlet_test.py,sha256=mCHvU1_l_2ZCV2S7Gk5jkwYWocQwObLDcOYFEqUUlFY,10477
distrax/_src/distributions/distribution.py,sha256=rTcOm7lsxa5og2_nRA4egreZq5VuQ4y2Ypgms9aKeuE,11822
distrax/_src/distributions/distribution_from_tfp.py,sha256=69S1KcoHzU-N5KMhtQIyr_qt9u0c6ajzzRQwNVXLl00,4765
distrax/_src/distributions/distribution_from_tfp_test.py,sha256=Wm07OWgCj8TNX0IIvLP1RIeZEQqahWZUpJP370VFK3I,9828
distrax/_src/distributions/distribution_test.py,sha256=Cb31oyA8BMRS6rZW2fpMml2vctgJXGFDA1k_UtYylKs,10098
distrax/_src/distributions/epsilon_greedy.py,sha256=CkodiXTl74w3m4EaPbEz6Q6kNbTVP8gb8UZMBPxkO7g,2888
distrax/_src/distributions/epsilon_greedy_test.py,sha256=0jkK2moqxucuLu79meVURbS-hTHt0vCDxFCZ8ECQeh0,3481
distrax/_src/distributions/gamma.py,sha256=QWB2cvApXrovUgWWD4I76bxGoiKYqiyjMyoFl4zh9Lk,5281
distrax/_src/distributions/gamma_test.py,sha256=5e4sKlfpK0ix3EJ-2MisQOR5al8LvNCUyUfigPxwqo0,8403
distrax/_src/distributions/greedy.py,sha256=j-fmV6sxNmjEx1zC_GYcjO-5y5nOJrdHYz4eknvszE4,2286
distrax/_src/distributions/greedy_test.py,sha256=6a7jaEaAJwViePXN0RDgVePOrKzBxKBIymUu0yIEmlY,3025
distrax/_src/distributions/gumbel.py,sha256=G_fAYf_Wd4E8BpWPw3ZG_Vrfpi6LrplMImlUgR7WIi4,5251
distrax/_src/distributions/gumbel_test.py,sha256=q13rhmVREk8ltFzh6DYmpOH90XxxC7AHLUBPZXzbZVo,9401
distrax/_src/distributions/independent.py,sha256=on3XJ8XCcDHNOeGQI4hQU4Ql7TpBy2_lclFKJyMgAY0,7145
distrax/_src/distributions/independent_test.py,sha256=Vgg0fJZCtU7bst_SkRxP1eHPzVjTTA-LEU9R0PHJyKA,16214
distrax/_src/distributions/joint.py,sha256=CVx1Qk4fV-TJzkKFSlg929xJ01hsK0NJUfmKpc1KGSU,6079
distrax/_src/distributions/joint_test.py,sha256=Ko3awCY-Yq3XsqXu8dBqfu1MtrCSpSY5_nELd_Vtf5M,18728
distrax/_src/distributions/laplace.py,sha256=DH4OVpiCiTPBx6mp4-DR7V2H-k7U08FMeyB63j_bPks,5459
distrax/_src/distributions/laplace_test.py,sha256=RB28-Amki4aUAKOqJDb8cyKNcEwuP1UIEN6r6CuAum8,11436
distrax/_src/distributions/log_stddev_normal.py,sha256=Sq-IB3pUHRdQ2gXuyGvrqx-8e1NcbrZgJlz7IXFa22o,3977
distrax/_src/distributions/log_stddev_normal_test.py,sha256=ck7TcjCB08FCTMlgc4YgoFRSadG3PajzevbqM0iA3So,7198
distrax/_src/distributions/logistic.py,sha256=TKgRlaiVMVsvoxBxlC-fR7YT42Asi6Jlyi7qPdRxhJI,4319
distrax/_src/distributions/logistic_test.py,sha256=b3EKDv4Sop7AlwRQ19ziNlb5aalUhzQ1ODAu-I4MV6Y,7270
distrax/_src/distributions/mixture_of_two.py,sha256=Ngt73MmNpk7DzGFoOJvlilFiM5Djym92Qt_loTc6ejo,4130
distrax/_src/distributions/mixture_of_two_test.py,sha256=W3rqtevm39C0jj9uhka5i-N7Oe0zMGUEjWt8wJk95Vw,6448
distrax/_src/distributions/mixture_same_family.py,sha256=RL2CDXUwX6-JMZKAOw8VLlep9mTbWyqfpjOBA3bCOno,6671
distrax/_src/distributions/mixture_same_family_test.py,sha256=LBNhYifwMwWwg7K7VPOMmlZ7Qega5ecDFFN3Z-Uhypw,12708
distrax/_src/distributions/multinomial.py,sha256=xT8Cn5cDuwoCJmB_lE6AtbSVnzAmL_18RCPbccxHKo0,11437
distrax/_src/distributions/multinomial_test.py,sha256=1aDQcG59SU6VW132I8AujXnHgOxzfOwcCc7w085RZwQ,21199
distrax/_src/distributions/mvn_diag.py,sha256=46SyrHk9ypYR7-bN_Tl6hBPDiivwQoLY9S4am-_5VDw,4727
distrax/_src/distributions/mvn_diag_plus_low_rank.py,sha256=xoToY-0oMAZb2IsMvK3dVyd8EMrLG_8xZYpdJYXpKmU,8441
distrax/_src/distributions/mvn_diag_plus_low_rank_test.py,sha256=Vw85nmSAccaeVoW80vp_B60KdwrcfKMqSdP9rFVXmRE,15802
distrax/_src/distributions/mvn_diag_test.py,sha256=2Cwe9U8PuuF9QgIf_vZ343AosYVg2CUgxqkljLH6QE0,17753
distrax/_src/distributions/mvn_from_bijector.py,sha256=N23PhOFnEFkyLr5vBjAZ-3N7BTfBmRmJxudX3kb1yhg,9763
distrax/_src/distributions/mvn_from_bijector_test.py,sha256=V00fuUsucyHhbDrTh0iqbtjIHS0TN2GUjOnNF3OnzSo,11596
distrax/_src/distributions/mvn_full_covariance.py,sha256=LuqH9FAhNMtTp-SHFNIG8jmEyNU60A-p0_P-1hDo-5E,5028
distrax/_src/distributions/mvn_full_covariance_test.py,sha256=KxaZcechM1HNmRZzki1BMf2DhaqgxIflFti77LfFaE4,8854
distrax/_src/distributions/mvn_kl_test.py,sha256=l4nUtzRvhPdmLBZz6Nmm2nFMYHRT4KjVB_TjPQRa4i4,5329
distrax/_src/distributions/mvn_tri.py,sha256=90kkrrcDq5mgkezDQIHSR-CtEDfojnNwLTz72nZYxe4,5227
distrax/_src/distributions/mvn_tri_test.py,sha256=TPjZJl78I86NhakG8UjmNl8wmmdxsQkkzuFbpSQ9XDE,10705
distrax/_src/distributions/normal.py,sha256=I4_OwB9KRDD0vwa4l7zwu_7Hgm_lxI5lGQa52Th9joI,5734
distrax/_src/distributions/normal_float64_test.py,sha256=e3z9rVl6TTMzaP2EDMo1qmxXoaYu88BmiOPPnxjyLi4,2677
distrax/_src/distributions/normal_test.py,sha256=85jbQsOzvIYcWzxkCAqJ0VMyGDymZf_cpempJsJRxMo,9253
distrax/_src/distributions/one_hot_categorical.py,sha256=_rWDBIVHEPfduZmCM8aOstNcPq91gUAtxdFvGJw77f4,3617
distrax/_src/distributions/one_hot_categorical_test.py,sha256=nVWda9oruojkCcCt59w3bfh1g0cnaiuuo-cpVi24rx8,18970
distrax/_src/distributions/quantized.py,sha256=nwqBguQvP5LRf0N2GsBkDMKWsWVkEzX9wZaBQRPV44E,13289
distrax/_src/distributions/quantized_test.py,sha256=hFqGd0z224tujLmkVnQQ5IE9lH5LhZ8d0Kou69LuoTg,17608
distrax/_src/distributions/softmax.py,sha256=LWlLdt-Cz9BBDkanJmETaRv1HrORfzsSF275GvaqduE,2397
distrax/_src/distributions/softmax_test.py,sha256=To0tgDYUdeKePv_Hpd_I_8ykBIerNvoC6_mo2B8bSSE,4051
distrax/_src/distributions/straight_through.py,sha256=3rBQUrsyHlOfgzrwcVFrufcdqcD0sX933ux2FnedN7I,2488
distrax/_src/distributions/straight_through_test.py,sha256=O3is_t4bslY4xgOMMz9fQm7ksTtE0U0oJ2Qy2-4B4oo,6977
distrax/_src/distributions/tfp_compatible_distribution.py,sha256=A5x9ZQnRfpFEYWW0Lw6RqYg5l7ahKjdfvnAAH-PzuIs,5917
distrax/_src/distributions/tfp_compatible_distribution_test.py,sha256=K_CVI6K4-BZ-Cys604acdNDvoYA2sq4A1BaUczVdJi8,11323
distrax/_src/distributions/transformed.py,sha256=PDkE2qawD16J1KFgBGULhyYW26hgyVzPBp53PH-xO8o,12331
distrax/_src/distributions/transformed_test.py,sha256=QS63Ue2ilmBdoc5U_R54ZP57G-TwivDRVgIlpf7tiVQ,19258
distrax/_src/distributions/uniform.py,sha256=IwkF6ZfUBrhea8N4-i0VHOL7ffMjrTVUbFLod-SWmEs,5286
distrax/_src/distributions/uniform_float64_test.py,sha256=kjBByYDGIFPHY7hTh7XAtJs_qJOaP7whSQzVoPyv8Z8,2665
distrax/_src/distributions/uniform_test.py,sha256=-M7rUujsYayCHXC96Rp6j_iyac26uTGoW6oRMMYz-ak,6351
distrax/_src/distributions/von_mises.py,sha256=4RLnrIc4WQdInDAWYVSq3d2BCQTkvQfdABc3QhGe1cc,17000
distrax/_src/distributions/von_mises_test.py,sha256=kh2Thw6NNPYTdUOVCRixQqftEkynwHc3Ud7ZcAHnqxo,20921
distrax/_src/utils/__init__.py,sha256=IV2YQAXXaG5v_BQRi14erjv-eiLmBA0Vauu42sj2yIA,696
distrax/_src/utils/conversion.py,sha256=hM0WLyNUlzvxI3FBlhtWvxlDI7UbATmjpjmlBANjc48,5141
distrax/_src/utils/conversion_test.py,sha256=xTNETan_KFHjnq-F-csE7Iv0wDU2Y7aimW4YH-sHFt0,6602
distrax/_src/utils/equivalence.py,sha256=iK_waRmdwu3pXvEtwQc07jwH4jQe57-0SRyFJFiY4Gs,14630
distrax/_src/utils/hmm.py,sha256=EUATUEOfqpere6DoQKv90NRUDbh_P1wYmnpdSk29YL8,11202
distrax/_src/utils/hmm_test.py,sha256=kMKUNZLjVuxRu1050MuDW8AnunuTViTw1Ixlb0wZVms,7354
distrax/_src/utils/importance_sampling.py,sha256=K_E67b-NDR-c8O71oX1C28wjzTaNwornkOHUB8rd730,1415
distrax/_src/utils/importance_sampling_test.py,sha256=sKCyxPGYHhGgH8Z8nMF1Pg3BpWvlhBQsoUcaKeC7XiE,2351
distrax/_src/utils/jittable.py,sha256=IBA7b4tMKIzO63dSAWLYapcDY0PmN9pOGc_uN9VOXC0,2590
distrax/_src/utils/jittable_test.py,sha256=J0FqZ4Zlh3NWVCHD4is9-lLc_A5400LnJfLZ-TsS2go,4020
distrax/_src/utils/math.py,sha256=-5I8GME6os442LlLBR2vjQQoDqyL_4kNJe4j1jeJGDI,4881
distrax/_src/utils/math_test.py,sha256=XyHuMnhNVnTvXdLB7t8dtvAKh1iraAnOBJXZged5bNU,4325
distrax/_src/utils/monte_carlo.py,sha256=FSZeNtRIrZOncpmy2ACC9lkaPIjer7-Tr3B4GStDDqU,5539
distrax/_src/utils/monte_carlo_test.py,sha256=55UKmqNgMCZqorsrsWpGoE3pqWDLIvIMCwXSWKFRUoU,4635
distrax/_src/utils/transformations.py,sha256=BR2ZqCTZa8xNglRb3CCqA0mS05bP8dxBk51erTdgD6w,12689
distrax/_src/utils/transformations_test.py,sha256=MB4X_U6pMn4PHcg1X_VvsgLRrMd6BSWllTPn7vVB0Bw,7340
examples/flow.py,sha256=jIPGNjX3CY9DY5qFCGw-7wxnUdBXqMI9yWnGSVM81_Y,6549
examples/hmm.py,sha256=byYRYpWhBegkOdhoG2Q3QXhsgA4howeLGhJpRJRTZfQ,10805
examples/vae.py,sha256=0Q0jNi3_IrXXzab8M2FhSQlWtKvB7jqS1RnG6seT6MQ,5949
requirements/requirements-examples.txt,sha256=xS2SxEabJCWLblOk7qGKApc8fjNpE82P585_TYiyWNs,74
requirements/requirements-tests.txt,sha256=1VsbpEN0fAROGpqtuLHOd9EMccKTPP-XbSYvTFPnTLo,28
requirements/requirements.txt,sha256=eK507p7hx5JgW9MWRdveOnAHeKAOC4alSUBhQ9YSLCM,99
distrax-0.1.4.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
distrax-0.1.4.dist-info/METADATA,sha256=XxiO1lnNE-9foWTjsMUkKeV-9egWFxmkiRClHNNF9AA,13208
distrax-0.1.4.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
distrax-0.1.4.dist-info/top_level.txt,sha256=duimu4ohJBPg5-RL7uAquTKOjYoVJ19LrnLVIswzFRM,30
distrax-0.1.4.dist-info/RECORD,,
