sbiax

View on PyPIReverse Dependencies (0)

0.0.55 sbiax-0.0.55-py3-none-any.whl

Wheel Details

Project: sbiax
Version: 0.0.55
Filename: sbiax-0.0.55-py3-none-any.whl
Download: [link]
Size: 4121971
MD5: ed5b9cf28b054c9f42785a6c1d571ed1
SHA256: 403c13a1acc95299e1c3a77742f77c61f64205da3674d7f1000e22f527bdbdd9
Uploaded: 2025-01-17 14:43:04 +0000

dist-info

METADATA

Metadata-Version: 2.4
Name: sbiax
Version: 0.0.55
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/sbiax
License: MIT License Copyright (c) [2024] [Jed Homer] 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,machine-learning,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.10
Requires-Dist: beartype (==0.19.0)
Requires-Dist: blackjax (==1.2.4)
Requires-Dist: chainconsumer (==1.1.2)
Requires-Dist: cloudpickle (==3.1.0)
Requires-Dist: diffrax (==0.6.0)
Requires-Dist: einops (==0.8.0)
Requires-Dist: equinox (==0.11.8)
Requires-Dist: flowjax (==16.0.0)
Requires-Dist: jax (==0.4.33)
Requires-Dist: matplotlib (==3.9.2)
Requires-Dist: ml-collections (==0.1.1)
Requires-Dist: numpy (==1.26.4)
Requires-Dist: optax (==0.2.3)
Requires-Dist: optuna (==4.0.0)
Requires-Dist: plotly (==5.24.1)
Requires-Dist: tensorflow-probability (==0.24.0)
Requires-Dist: tqdm (==4.66.5)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 5020 characters]

WHEEL

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

RECORD

Path Digest Size
.gitignore sha256=xmbOiuNkvaHbxIyxcUHv74ONERqgFswFJf-t-WCbIcs 538
CONTRIBUTING.md sha256=_W_k7WP-L2_HPz9qJI_WEiKqPc-9L1jJxn-INlzgB8o 699
LICENSE sha256=Fdl12BxdEa6xkbxTlGe2gwYCjX2bgweYqV8zCgpBhbk 1069
PKG-INFO sha256=F1ecTLF_sVokSlroqakkHSq_WygoXdkARe42z4Sgvzc 7553
README.md sha256=IfIoOJ3J1ttgFYsNQNrj3Qy_8UDjQJ9J7w-GDKF5MNc 5020
pyproject.toml sha256=He4-K1qh2d52AksGlDYWVLriQq_piHu5h2BMHRxxEAA 1345
.github/workflows/joss.yml sha256=CvB6w4L53tdWY6Ahm24uAYEOtD0R-RtPXx6_OEL3zLU 1221
.github/workflows/publish.yml sha256=WiGsl7TA6XEqYtFYlCWyWorEn_6yzb4vUjj57tjkM6s 26914
.github/workflows/tests.yml sha256=cXDXKVigGXCG_oxgtleaIu0XHFJBO84tDijDSFpPOfk 795
assets/cover.png sha256=OmGMmw0MdTmB0WcsmhxXfVH2oV_Nl1egu-5Z4qExTS0 63904
assets/cover_dark.png sha256=4VmG7pG3TNI5LbNBG2RNhNem7fmCJMEzXAo5HRHWJsM 54011
configs/__init__.py sha256=6PzHFwS_dST_ZkmNm4f_bT2Z0t5uCgyrGuh2OHXvw5U 1167
configs/shear.py sha256=b_1rUDQFNX720My2YzMn87NsWaGvcqErD6y2VmO5MaQ 1017
data/shear.py sha256=eGOWac93Zvqw6B7UEV3ivnxkvzsRetRX01EBsgWxkng 4938
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=2sRNIacwHa12SVaY4AGQgrIQAfdBz9L8Z3kLErblPVE 461726
examples/shear_nn.ipynb sha256=eMRbMEIL1EmeYoSvsTjqFiC6kHRS_iH2lmNu-kLwahU 905657
examples/shear_optuna.ipynb sha256=iSaDG7vDR6plmT6apQ4U_JNV6PYltgI4Gxrx557YcIo 220268
examples/toy.ipynb sha256=NdekSjBHl_ySyJCNixvcCEyMfsKg__NCS81GC1xissA 265173
paper/paper.bib sha256=Pbo8B_yK0bNHm67I9GeW6qd31nOq5ygVDAWFkpxU2s4 15965
paper/paper.md sha256=JNP9IdeR54dXGFqq13ej1d9zEoaYlCwfQIkyww9fomk 11929
paper/paper.pdf sha256=1NwDr73EIsBXm1rGt0rFxGj-xYr_nJcI2dv_v3IJMnA 366958
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=z86kmKvtpXzUSiTAYP72UgO9VkXAUqfoEdEz5Q_F6ZU 3386
sbiax/compression/nn.py sha256=Gzp3p_DEQn_EFGN6LhszZUGQoyrfEkkqC4qswi8VF30 6157
sbiax/inference/__init__.py sha256=nwSWcRZSoNAaxkD2eHyOwfflsALb9Qye4L-drcWua18 29
sbiax/inference/nuts.py sha256=7gK85OIprenv71zHKZbEuKHHA6id52Wvhkpk-4Q2M7I 7543
sbiax/inference/nle/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sbiax/meta/__init__.py sha256=R0lg5f4YMNWMH4IrhszW4okCvcf-NhZE3RDry57ud0U 56
sbiax/meta/_optuna.py sha256=QMrMzmXw6z8NO2BsbnGaRazbXEA7zgfeA5qxdSWw9rs 7387
sbiax/ndes/__init__.py sha256=LHz6OfR_2Fx3U0-jNO2VXeTFcpqkF3lFRilOwKiS6qI 120
sbiax/ndes/cnf.py sha256=06i8JBf7RXVto9E_IEQ7EmDwSp4Vcy3FDnak4Zggw00 22376
sbiax/ndes/ensemble.py sha256=eB4211m_BlFOj1z3qlu0-TUGXg0-4ZvAf-eiCI114eU 8945
sbiax/ndes/gmm.py sha256=2l-dp6Aois4TCsaXA4wTm_le5VjjB_eJzfqSkDJ1GMc 11409
sbiax/ndes/maf.py sha256=UVIDro_kMuMj1dBWRoC-g-xgcUwoY7IZ2MVN69akdpc 8635
sbiax/ndes/scaler.py sha256=Jv-V-8qzC2CXzH8Dq3g8M1V8pK2wZnnvxiPtsw_8lKc 4540
sbiax/train/__init__.py sha256=E1ydCBRke9E-oyQ4a4c3_s_88Iyn8xIlPxTS1lgJle4 44
sbiax/train/loader.py sha256=TH7vHM2U3LBr9c_yhbyviE5FMpZsXXgdUbqq4MYJLMA 6668
sbiax/train/loss.py sha256=KoNV29e4kun80KR62l3yTqXFyb72EU9wZCpAywPe3y4 1421
sbiax/train/train.py sha256=d2axRs4sT44EtcwfRFdfFUPJFmNqqhX2_3zDJ2UUOxk 25619
sbiax/utils/__init__.py sha256=2Y7It--11hsQgZW1H-5IiPAtdIS7-EyATH9J28-IUWk 1467
tests/requirements.txt sha256=wVDY4RE3DyHLRZdJ6gI3cvt4mMHjF7MrYWYdOyBK_qQ 30
tests/test_fit_nn.py sha256=DcsVFJblYH4jXl_-mPBe5Mj5vqcy85lBXKm3IcZdb7g 873
tests/test_ndes.py sha256=6MzCb8GFDi8l98YLDNeIsiCen57Do-G5j5adNVlxq5g 1675
tests/test_npe_nle.py sha256=-EqxsQKWQpqjfvTpi9mxHzq-ZUaLFvIgQ6gMxGGDILs 4017
tests/test_nuts.py sha256=hAPWn1i9JG6Rz1WZhkp-xc_clWjFODHkLf1HPVqkMEQ 1869
tests/test_toy_example.py sha256=IMK6iSUzDbeCNdnRWh0X24j0rux5IEKus5r4XqDqd-I 10775
sbiax-0.0.55.dist-info/METADATA sha256=F1ecTLF_sVokSlroqakkHSq_WygoXdkARe42z4Sgvzc 7553
sbiax-0.0.55.dist-info/WHEEL sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ 87
sbiax-0.0.55.dist-info/licenses/LICENSE sha256=Fdl12BxdEa6xkbxTlGe2gwYCjX2bgweYqV8zCgpBhbk 1069
sbiax-0.0.55.dist-info/RECORD