sbiax

View on PyPIReverse Dependencies (0)

0.0.12 sbiax-0.0.12-py3-none-any.whl

Wheel Details

Project: sbiax
Version: 0.0.12
Filename: sbiax-0.0.12-py3-none-any.whl
Download: [link]
Size: 3522558
MD5: 5d6721b0a4c81e41934aa2940dd04bd0
SHA256: b8bdd5aae6caf254ca83fe04a7edc8e5a5d4786ee0761c9c9c2f51d425b067b7
Uploaded: 2024-10-27 14:30:06 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: sbiax
Version: 0.0.12
Summary: Fast, parallel and lightweight simulation-based inference in JAX.
Author-Email: Jed Homer <jedhmr[at]gmail.com>
Project-Url: repository, https://github.com/homerjed/sbi1pdf
License: MIT License Copyright (c) [year] [fullname] Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
Keywords: bayesian-inference,deep-learning,generative-models,jax,likelihood-free-inference,simulation-based-inference
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: ~=3.12
Requires-Dist: beartype
Requires-Dist: blackjax
Requires-Dist: chainconsumer
Requires-Dist: cloudpickle
Requires-Dist: diffrax
Requires-Dist: einops
Requires-Dist: equinox
Requires-Dist: flowjax
Requires-Dist: jax (==0.4.33)
Requires-Dist: matplotlib
Requires-Dist: ml-collections
Requires-Dist: numpy
Requires-Dist: optax
Requires-Dist: optuna
Requires-Dist: plotly
Requires-Dist: scikit-learn
Requires-Dist: tensorflow-probability
Requires-Dist: tqdm
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4387 characters]

WHEEL

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

RECORD

Path Digest Size
.gitignore sha256=6TQi0OOJHaqK4LNvcSPv1CONsWhHeSXGAbGxMc5BuOc 200
LICENSE sha256=pAZXnNE2dxxwXFIduGyn1gpvPefJtUYOYZOi3yeGG94 1068
PKG-INFO sha256=pxjLs7gBHTaPtJNC1IBzdAgzUH1SkFQnEqDYWATZeaA 6813
README.md sha256=ZDkYKEaiQoLBKfQMrkWV9L1j36tcRdV1yg2NR44Ne0Y 4387
pyproject.toml sha256=frJ0YZdBf8rLZCoRBWBB3AE6ibFveUyLedqITpJkV8E 1224
.github/workflows/joss.yml sha256=CvB6w4L53tdWY6Ahm24uAYEOtD0R-RtPXx6_OEL3zLU 1221
.github/workflows/publish.yml sha256=asjGIpOr1FuEd8BYuSJwGhSNjvQ1ybjY9V92g4W1Xtg 3368
.github/workflows/tests.yml sha256=LOTyLL7wa2A2N2EfsTHIIOM-KudEhRn_f5O2thy7ueg 727
assets/cover.png sha256=OmGMmw0MdTmB0WcsmhxXfVH2oV_Nl1egu-5Z4qExTS0 63904
assets/cover_dark.png sha256=4VmG7pG3TNI5LbNBG2RNhNem7fmCJMEzXAo5HRHWJsM 54011
configs/__init__.py sha256=RMbiw6kNpQ2MxESNPwMbcK296XF-hwUpF45gfmXWJuk 938
configs/shear.py sha256=b_1rUDQFNX720My2YzMn87NsWaGvcqErD6y2VmO5MaQ 1017
data/shear.py sha256=eZoeAOoG44_8S_nOwiwoor6qyRk_3AKZjP0nIfuv2uc 4999
data/shear/DES_shear-shear_a1.0_b0.5_data_vector sha256=b065rHdBJpWl4YNERlMj169B9cXXIJKZwyduc0Xj7pc 22500
data/shear/covariance_cosmic_shear_PMEpaper.dat sha256=mr2o40YNApV2rNVPFo23uvDQkMkeHqwUrFLOev5CoX4 5071528
data/shear/derivatives.dat sha256=uebbXquhtdEKY3vms6Q459IzMOw9IxZ1oglD2TqYczA 34396
examples/shear.ipynb sha256=bc0TJDoeClyFV2psOMWad1ZopahQ81Rj0_X1sl_ndEY 444114
examples/shear_nn.ipynb sha256=DXrVkdlmFBzV6FezHtG2-ytbjrMkDtb93mwuWJPFSOM 464538
examples/shear_optuna.ipynb sha256=wV6-vG1LitmIvKBzDqnoMc7anBUaV6Sgz6lcojh4Gfc 156536
paper/paper.bib sha256=kyNJMlFJXWuTWmtGziNtN1BjjTulUKom3e-99ZAGGSk 6209
paper/paper.md sha256=b9mitFSQ1z4-gsOTv_pAqecc4vh4sMlJd2g0LtJYqoA 10772
paper/paper.pdf sha256=m7V-OEuaIM4S8OLUTbMSBfi_qSJ33R_HPJTzAbZk1X4 363619
paper/sbi_example.png sha256=XhuUoHrPUCgZp-Yf2e-_tLk90CIw3VVQv90ksZIuFK4 112968
sbiax/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbiax/compression/__init__.py sha256=UeiF0ou_AOX7xFent8m5KjQKstsGn8_DQLkzVWmg8zE 52
sbiax/compression/linear.py sha256=1nJZIT7EL2jSZhTwU4M68Noie01raB4pHX6uQfjvuOQ 811
sbiax/compression/nn.py sha256=WTos9OpBArDNdFjsrvqR5TiOBLayaFh6iJEvRg3pzfI 4151
sbiax/inference/__init__.py sha256=VW1ia3IbE8E3l3hrDDN8v_tZVEzzFExf2hpbfateuD0 134
sbiax/inference/nle/__init__.py sha256=Wqz5N2bZYVCeQltdptXVBxaMnY1cPpcsnAuWTJ01rBA 129
sbiax/inference/nle/log_prob_fns.py sha256=95i79FsJqgBjeFHMMZJniwDnsYVvXlztOtgp4Yexfms 3009
sbiax/inference/nle/nuts.py sha256=zej2ZahZtS3ClB8bcRborYOLtCMuq4WMEBimClCxHL0 4722
sbiax/inference/npe/__init__.py sha256=nu2ErXM7JxyqDR3BfT_EfChDkVMbViq7TLTnoTj_NK0 47
sbiax/inference/npe/npe.py sha256=P8Ps5OATZcXIl0HzImsqPCq3dNMF89Co1Alg6tA6mgw 723
sbiax/meta/__init__.py sha256=SKRcEPoluRejs9jBpTeqd-lYVSxd-OHC1tx4bH_9j0M 129
sbiax/meta/_optuna.py sha256=aM830UOzqzJsAL3b1GXe6FuvRj3h9XYC8unzc8_nM2s 3344
sbiax/meta/_plots.py sha256=HKnm3S7a0170fAgfowruDSRjVsxUqUO9Q_o4pkACQ-0 2474
sbiax/ndes/__init__.py sha256=kuiL7pFloR3c9JVLLoy73B7vvXQ5kAsedXsb0TP70aM 324
sbiax/ndes/cnf.py sha256=b3mH6HjbV0WlbTKMzOpAxodv2bxGjXaovokPDggSQzU 22082
sbiax/ndes/ensemble.py sha256=6juw5eoJ3Ez6qf8Rc3hgT-Z9lmjGQLjmMmuYNr2SON8 8350
sbiax/ndes/gmm.py sha256=c7jaOYN146-I7cCudtrA9vqELBISqBzb2xuFdy3izKA 11179
sbiax/ndes/maf.py sha256=_ljINpPy5t5lutEZGFmGuA1o0pCYsMbW8B7cTsn7UMc 8423
sbiax/ndes/multi_ensemble.py sha256=8ea-pjk8PpUO7hCYtvvA9XDNIA0-RRGncHu9PJRV36Y 1394
sbiax/ndes/ndes.py sha256=gVoh3XM__sU2l-6PeaAQF7xF7J_HgfNiBNeuVwdz8CA 3738
sbiax/ndes/scaler.py sha256=5LoaC0OBRG2dZ121MB-ImO1-ZjB5vL94e9_8VOhhHpE 1620
sbiax/train/__init__.py sha256=E1ydCBRke9E-oyQ4a4c3_s_88Iyn8xIlPxTS1lgJle4 44
sbiax/train/loader.py sha256=A4iLZOsiY2nmMW-_xY0eDmP5LCAStgC4J-tmlZoBCuw 3198
sbiax/train/loss.py sha256=P-gMpsACP47fflRmQbbgb4FKbMg8BIGmWEx9pxmUJSI 1029
sbiax/train/train.py sha256=FbnGNPQH7YKLSQYVobvOZCdeR9d3ZtKua5LgPL_bRgo 18923
sbiax/utils/__init__.py sha256=-CtDDP2ieuJmknUfqW5TI60ZdKokuWoysfq4YYWvO1g 376
tests/requirements.txt sha256=KwEhTTLDgtvHO_a0k77beSYySvJ7rT7oiPZndLmxFMg 6
tests/test_analysis.py sha256=0j6ESwrQViuVfHiqlIgei1b-35YUcIqCxZvfreT8oYM 3706
tests/test_ndes.py sha256=9N3uWgVj6tWd1jaS8ES5uP8S7sAZ-FzK8JF91Xg0qgM 1483
sbiax-0.0.12.dist-info/METADATA sha256=pxjLs7gBHTaPtJNC1IBzdAgzUH1SkFQnEqDYWATZeaA 6813
sbiax-0.0.12.dist-info/WHEEL sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY 87
sbiax-0.0.12.dist-info/licenses/LICENSE sha256=pAZXnNE2dxxwXFIduGyn1gpvPefJtUYOYZOi3yeGG94 1068
sbiax-0.0.12.dist-info/RECORD