sbijax

View on PyPIReverse Dependencies (0)

0.3.0 sbijax-0.3.0-py3-none-any.whl

Wheel Details

Project: sbijax
Version: 0.3.0
Filename: sbijax-0.3.0-py3-none-any.whl
Download: [link]
Size: 64254
MD5: 041ac3ec0894459abd2d13dc90d067c6
SHA256: 6561755757f1ab67c318f161f063ae45e704de258b75967c104378baba4a98c1
Uploaded: 2024-08-18 12:47:34 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: sbijax
Version: 0.3.0
Summary: Simulation-based inference in JAX
Author-Email: Simon Dirmeier <sfyrbnd[at]pm.me>
Project-Url: Homepage, https://github.com/dirmeier/sbijax
Project-Url: Documentation, https://sbijax.readthedocs.io
Keywords: abc,approximate Bayesian computation,sbi,simulation-based inference
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Requires-Dist: arviz (>=0.17.1)
Requires-Dist: blackjax-nightly (>=1.0.0.post17)
Requires-Dist: distrax (>=0.1.2)
Requires-Dist: dm-haiku (>=0.0.9)
Requires-Dist: matplotlib (>=3.6.2)
Requires-Dist: optax (>=0.1.3)
Requires-Dist: seaborn (>=0.12.2)
Requires-Dist: surjectors (>=0.3.3)
Requires-Dist: tensorflow-datasets (==4.9.3)
Requires-Dist: tensorflow (==2.15.0)
Requires-Dist: tfp-nightly (>=0.20.0.dev20230404)
Requires-Dist: tqdm (>=4.64.1)
Description-Content-Type: text/markdown
License-Expression: Apache-2.0
License-File: LICENSE
[Description omitted; length: 3930 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.25.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
sbijax/__init__.py sha256=2I4QfewtVIbhIcVJUjaASTqgLMxCxOmuV4HF4My7RDs 1211
sbijax/_src/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbijax/_src/_ne_base.py sha256=AB-zPvoWVMGEgiR4Lfw1yqzci4VDar8Bu_p6UascVwk 5915
sbijax/_src/_sbi_base.py sha256=47uZmIqBqBAxXIB4eRM0NzgFNAA20gR-S-sdWBRnZuQ 1285
sbijax/_src/cmpe.py sha256=LnlGE1BcYdjuErEXDzbmFbp8UN0CuWhQwSmi5BZrNmY 7254
sbijax/_src/cmpe_test.py sha256=PUWCnPSnaihArrK3O1DJmfdShljyrQrENpzA6V-0Fe8 871
sbijax/_src/conftest.py sha256=ERHKBZvkaM10T1WNKBr3BHe1NUOExMmWxSQxzvmkWSs 549
sbijax/_src/fmpe.py sha256=i50xl-dPt8RLQEwC-1PGbIJsoaRl6h9iTDlpFffLMdU 9469
sbijax/_src/fmpe_test.py sha256=S6v6mhk84GSz6Pwzyyk5Co0xtV-ILvytiavWy2Jid24 873
sbijax/_src/nass.py sha256=9clARS6VryPRESttXYmQgH9Fd3N2nQ9UQZuZtSC2ZVE 7822
sbijax/_src/nass_test.py sha256=exFQgvS5cvhTIg-2JwSRXwIsIF5uIOd4y8QSQUAHKGU 1518
sbijax/_src/nasss.py sha256=BLz_jbp6canwtTsorO5WN6-BcNuG76dHxehUJXjLLCA 5846
sbijax/_src/nasss_test.py sha256=be6fWGgcTRf5502s7DGWYA_XCydAZDICnQB-XsQ9NVg 1549
sbijax/_src/nle.py sha256=4wx4vRVSGHwh1mJYoeU68RqPP1ioJX9fcrtbeFJo-iE 11601
sbijax/_src/nle_test.py sha256=n2ynsqvt6-s3qlAP0UndCZ3EwZWe3Ww1D6czH8Ad71w 1208
sbijax/_src/npe.py sha256=FHN8Zum-rr2-6ijBhRL5xLC-GXKRpXU5bcXikf2a5jQ 11112
sbijax/_src/npe_test.py sha256=QJSN3qPWDYEAwWhMih3dS9bqO-CDmndSGT87e5NkHE4 828
sbijax/_src/nre.py sha256=TvaMMPn7YyKKvsCjVJMbHDbo59e8mgBVqF2ZaVdrSRI 13965
sbijax/_src/nre_test.py sha256=kDJEZqDF1BsemLJ778zB-tJv6DIFs83_XeF-Mbn0xnI 835
sbijax/_src/snle.py sha256=MsZY61hED-ipHgO7VAKotV6OKVQaZ-nfLHx2lWIoq_I 1510
sbijax/_src/abc/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbijax/_src/abc/smc_abc.py sha256=ZAVfpmUnL8XFZxlo2c4lTCd91F4_fCAUA_uTh6Za4RE 9367
sbijax/_src/mcmc/__init__.py sha256=wtKGhQrxjhFU7_ouDTJqT-9nxuajjsI-niV3NcDzJn0 463
sbijax/_src/mcmc/irmh.py sha256=r4shUeZ65oae3e-0JV5PyMIQbfVxZpNYzQTQqcY2b_U 2035
sbijax/_src/mcmc/mala.py sha256=9eCbIhGWaTTGRq7Bt3_VrCqvs5vBAnUdwz8o2nMg3_U 1912
sbijax/_src/mcmc/nuts.py sha256=hHb9zWGPfzfN24N0ZBpAb3IW4vdWxcxBAssQSJ7twRQ 2269
sbijax/_src/mcmc/rmh.py sha256=-nlgf2kfYTvskDGgr_QyEJUERvztGteIpywEcYSxBok 1977
sbijax/_src/mcmc/slice.py sha256=oQStt4HYrd1yY482oS0QHAhTz4L9Z4S-6J4j5alG8iw 1506
sbijax/_src/mcmc/util.py sha256=0dwO6faj1o9KF8fMgYNjjDjgvCS1bfcCDfJW-qNWMm0 232
sbijax/_src/nn/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbijax/_src/nn/make_consistency_model.py sha256=jckE9YTe5a2yXtu4FAQJ5I1BitDL5MLl-53moD-z5KA 6997
sbijax/_src/nn/make_continuous_flow.py sha256=iCEue1oUMfSOhX1b_NYWlDcKUMRlpK74q3OR8k9VTLQ 6159
sbijax/_src/nn/make_flow.py sha256=4zqa37scDg4wU5rtZW-reaCD5iBWWH2Gf1iXVT_6K7Y 9930
sbijax/_src/nn/make_mdn.py sha256=PylseT4fq2RM4byJ5G-K4o7dSxwzEL6po1YRP99cO2o 1810
sbijax/_src/nn/make_mlp.py sha256=IBMW4KFd9x6RvPsnuvPxC6vl5RwiPT9Qhl8H5TSYsBA 887
sbijax/_src/nn/make_nass_network.py sha256=DZSSw9bseIHw56yfcXMiuezDR169SUbRQCm7oCQAJuA 2458
sbijax/_src/nn/make_resnet.py sha256=dQwreg9_JuaHBs8z9huhIEvarrKHjx7znLGYKlFFYNM 3781
sbijax/_src/nn/nass_net.py sha256=S5i8F-m_LQW8FM70AKXIsqzOOhkChZH8MTeRKhLtoJM 2382
sbijax/_src/nn/nasss_net.py sha256=omjLsRVEIEnwRt4CZPn6q4HeK2ZoFH6aDYawrmFSzdM 2585
sbijax/_src/plot/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbijax/_src/plot/plot.py sha256=jFlATH28GtdJweXY214hdve36Duw4uLCNniIu-Wz8pM 7411
sbijax/_src/plot/styles/sbijax-bluish.mplstyle sha256=x-YPW6rwztXLbhAi8-q0eF8s8dNkGpjBNtOjLYRFkaI 85
sbijax/_src/plot/styles/sbijax-grayish.mplstyle sha256=_7qL6nChjFaNsDQUuWAeKRMj-Wk0rjBaueueviGJpSY 85
sbijax/_src/plot/styles/sbijax.mplstyle sha256=8uQKPC2_8mh1rBiOCrWSwRNsyzw7XCToR4GGjYHrR9k 562
sbijax/_src/util/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbijax/_src/util/data.py sha256=q-OapHDX_sLIsVEpgs3wf_rVAXArEwtq0jeZScadTgg 2071
sbijax/_src/util/data_test.py sha256=iwA4gBfd1BNTMme8w6PTJGjJQ1MWYS47cOVRoQbpxTE 1167
sbijax/_src/util/dataloader.py sha256=sSXtsjwxHAs1Zi1gVWt-BAo08SQxSiL_NO3ONlS4hsM 3508
sbijax/_src/util/early_stopping.py sha256=cU6QMXu-YNKGczEGx4LG630SWVCzg6m2pF1rvKg86cU 1168
sbijax/_src/util/types.py sha256=dEwk7w_Vhj19K0q6fr0q77c2Kddc-sjKQIqrlY3_uNc 37
sbijax/mcmc/__init__.py sha256=HKBwnTb_PAcKu4Rjm5B-3egeDzFBySzui-EhQ6UVU3U 410
sbijax/nn/__init__.py sha256=kszJsdBjVfxAYeL3dbrr0W5LLC9Uvgo01EIqlv6MpM8 610
sbijax/util/__init__.py sha256=QSoyGF-o8DWAhsy3Wui4abHWIiMarqzKC5F7ZnUMVDE 97
sbijax-0.3.0.dist-info/METADATA sha256=3nE5uBQGsRKCByOuvErObUAjshHFn3S65NyZ2fXQtss 5159
sbijax-0.3.0.dist-info/WHEEL sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY 87
sbijax-0.3.0.dist-info/licenses/LICENSE sha256=syDm1QE1qG4IlCDYjIr2IGzOpF_Kij6S-r73xYR_DcM 11345
sbijax-0.3.0.dist-info/RECORD