Wheelodex » Projects » probjax

probjax

View on PyPI — Reverse Dependencies (0)

0.1.0 probjax-0.1.0-py3-none-any.whl

Wheel Details

Project: probjax
Version: 0.1.0
Filename: probjax-0.1.0-py3-none-any.whl
Download: [link]
Size: 152203
MD5: a041d4445f544595ff7fd3fb0f0b30b7
SHA256: ae43839f886aae73a2e44a2d79f9009d4f37e4b49f47fa484325e243ed2ed5e5
Uploaded: 2024-11-07 13:57:45 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: probjax
Version: 0.1.0
Summary: Jax library for probabilistic computations
Author-Email: Manuel Gloeckler <manuel.gloeckler[at]uni-tuebingen.de>
License: MIT
Keywords: probabilistic,jax,computation
Classifier: License :: OSI Approved :: MIT License
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: POSIX :: Linux
Classifier: Operating System :: MacOS :: MacOS X
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Requires-Python: >=3.9
Requires-Dist: numpy (>=2.1.2)
Requires-Dist: matplotlib
Requires-Dist: jax (>=0.4.34)
Requires-Dist: jaxlib (>=0.4.34)
Requires-Dist: optax
Requires-Dist: ott-jax
Requires-Dist: networkx
Requires-Dist: blackjax
Requires-Dist: sympy
Requires-Dist: pytest; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Requires-Dist: pytest-xdist; extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE.txt
[Description omitted; length: 1515 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.3.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
probjax/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/core/__init__.py sha256=Yp0uy7qxTIfstKAoXWUZpxJssLwmVEnLqnmtzASNxAQ 345
probjax/core/transformation.py sha256=RLvFLZSARy0Iz7dJ9ipjn09HQWtm9iinwhAsZ-y-7I0 7698
probjax/core/custom_primitives/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/core/custom_primitives/custom_inverse.py sha256=h57EG0nf0JBn8_RK0I0Bj0dYMjH82xIcbBbuaiBrnjU 8502
probjax/core/custom_primitives/random_variable.py sha256=Hwg7N4cUGToVlJXsghFSDeIsL_0eE_LJyR-MR5jzHqM 7301
probjax/core/interpreters/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/core/interpreters/constraints.py sha256=cf_J2CG35v3AL94E-vACrvzUfSwOGi8i9QM9OuNot-0 1440
probjax/core/interpreters/interventions.py sha256=rBJtZiuUbWtVz-vVTx-JTIgDCtY_7zU-5sJzhLwFWYY 1388
probjax/core/interpreters/inverse.py sha256=UJgH7ZVLIcapelIrZdS7Dh_Ee8Lq9pJZCPn-MhyCFr8 24061
probjax/core/interpreters/joint_sample.py sha256=q33AQhyowIZIScrQYC8s9mvJ1RY6C0oRkDrsHYISCRw 1176
probjax/core/interpreters/log_potential.py sha256=0ESDu3_aniIod45JQb9XfnPcl6WhaHMS253ayJaJXOA 3150
probjax/core/interpreters/symbolic.py sha256=PFiVQlILIh3lUeGrqkssCpKkisvLLT8f8aLj2RrRDsI 3471
probjax/core/interpreters/trace.py sha256=GXemmawJRLJPX_jmhLXY9VvbHeVnNKigP2FGrxnJZo8 1213
probjax/core/jaxpr_propagation/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/core/jaxpr_propagation/graph.py sha256=7AGpjTQKQC2DrCqwEKwN2L5F-bjl8Q0VDRfQQTTHPIo 10661
probjax/core/jaxpr_propagation/interpret.py sha256=7qRpssRBmv2l0ASa6Uz8n3TPBRsB5eyeaftZ4479mb8 2440
probjax/core/jaxpr_propagation/propagate.py sha256=V-y2OBVWJvvUUCb43oFNieiPLmPlReyL-zPatlkzUJs 6855
probjax/core/jaxpr_propagation/utils.py sha256=rDagCCJ1Zyi5yEZwBIIfF0kwcu2ExbxzebYaPkUNBwM 3382
probjax/distributions/__init__.py sha256=DxtHs8iNWumONW1HIO981D1CwW4RRhnEZeXLSIvdTJE 241
probjax/distributions/constraint_registry.py sha256=ZpW_fVNXP700TkXx7a628GfNhFSWGiuMKz64IaGGVjY 5047
probjax/distributions/constraints.py sha256=Ia4Y567ktrc4v21bSBM7VMJWjlqhZtaUByzA2tA39xk 7699
probjax/distributions/continuous.py sha256=P-TGg4MnmJjeMYAnsbuRmM3tyCMSaLFZE9nRUuvx5W0 26603
probjax/distributions/discrete.py sha256=6Z6S6UFp5wUiwiEx9je7dJfQvetV-z-iNu4qDeov9wE 11701
probjax/distributions/distribution.py sha256=S-DayQcX66ukIkGN43pTppzzfW4ogWq0qvdYuRvdOWw 5774
probjax/distributions/exponential_family.py sha256=C6t4g15_JRfpcD4rEzvR-i2zXODOuPuWHrkaMdoRbLc 2442
probjax/distributions/independent.py sha256=TWn4SI3TPTNA3tl68BvAI8q7XZFUWPTWHk3krvmHBqU 9042
probjax/distributions/mixture.py sha256=MFeKV2sYOfNfxuL3UpejZGNMC97ux4N90QP4eoTbC1g 4993
probjax/distributions/sde.py sha256=FC0BIyuZJ7H8WA_xoRuIb8VwjZpxTIM2RG10hIizbhA 19836
probjax/distributions/transformed_distribution.py sha256=5Vbgy0JNllgtv-26rPkVa7GAqI0hSyE_fAAL3SH5pWQ 4217
probjax/distributions/truncated_distribution.py sha256=HZa8nDJEdNT06EXaYMnt9sJR069XmAJa4RWvWRJII9U 1883
probjax/distributions/utils.py sha256=1_lF8ixe-S6sf67pbOUiJVdbbUObMzUAamG4uhJajfQ 796
probjax/distributions/divergences/__init__.py sha256=gC7xiCAKJq6Ek9GnUHnx7mjkE3Ej91d2axABxmXzfTc 142
probjax/distributions/divergences/divergence.py sha256=gZbQOB_2_EKHyjBo_w4k3sxGTe7iuz8zH14dk9el81A 4370
probjax/distributions/divergences/kl.py sha256=NLHDzSp-1mygfe2smXIHTSh8DQ72bFygy7c9_JRHYpE 8804
probjax/distributions/divergences/wasserstein.py sha256=AqEn3Le6nlD-zf3YKZse7JZuAGg_iyAMx6JmzIJKhOY 6321
probjax/inference/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/inference/filter.py sha256=lhwJTttD4E1aGQkgN1QQ9CRDQrCcm8EP3nWWn1YenIM 2770
probjax/inference/mcmc.py sha256=FtCVXSJJnT8X18K_JkpHETR0cYaizLyJ7sMH-Iei_Vs 4542
probjax/inference/rejection.py sha256=PkBAF8XxTm5mnWS0QFh7TgDjeLTo11nqwFDefJsVd18 2675
probjax/inference/smc.py sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
probjax/inference/filtering/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/inference/filtering/base.py sha256=A9OrJoZR25gb5AnH3CUG6Go3jqHNeZZX6BZtE1XKis0 1380
probjax/inference/filtering/kalman_filter.py sha256=pPyI1OHzrQPzKKCIHdJ1xY0647InY2UopTJN25Pezqs 3697
probjax/inference/filtering/kalman_filter_old.py sha256=pOHNSvJpfeZ6SJiJMP9n5KUCDNXPaWpfPxcD4dI6LtU 8621
probjax/inference/filtering/particle_filter.py sha256=NSvsEuvQ9CqEgMHcWdWVxEAKKV0UTWrMWhAF-KIHexU 7117
probjax/inference/filtering/uncented_kalman_filter.py sha256=yvbhc7egLBwT3iiVRHI85FkfFy-I0xaUwzhWOyCIjvg 8754
probjax/inference/kernels/__init__.py sha256=PJr4-iRxP834qbLk2sq4Nvf0AR5vezK7_uA4U2_dWhI 453
probjax/inference/kernels/adaptation.py sha256=Yt5uHkMRS0F_NMYYlsNBgZFq7Cf0V17gm4UzTCWQWUA 6318
probjax/inference/kernels/base.py sha256=Ri76775xEQV_To4RL7LmMZYs1LU_dHgAVAwl6RH9RT0 2170
probjax/inference/kernels/dynamic_hmc.py sha256=KE8WK1cpmxjdbheiP0u0V2FCvYla2-74fY_LObvbo1A 3979
probjax/inference/kernels/elliptical_slice.py sha256=PWezZ7D94yyw0jBrZTDjlSvwxNQ7gJaSYM3UvGe7FB0 1026
probjax/inference/kernels/gibbs.py sha256=NZ9doaDXCFc3stX0v8yOHfpwgx32XOgDjVnB5VqD3eY 4118
probjax/inference/kernels/hmc.py sha256=S9p4bpcJINJsh7d0ofL8l04X1dwC4m-rteBHLY1RCHM 5526
probjax/inference/kernels/imh.py sha256=NJp_1Q10rep-5oCpr6SK2tKvGwbK-HEJIKBNfDgQIaQ 3517
probjax/inference/kernels/latent_slice.py sha256=Z0wQN2GMUcdaGLcY3j2Ww5bkGznhxZRd-CSU0IudHo0 5350
probjax/inference/kernels/mala.py sha256=jDf79jM7I8f1OlEx-NlEvQMzbfEE34IRFDn27Oy_59g 1680
probjax/inference/kernels/mh.py sha256=bz60nQ8OU5RRmcrzXA1tioirnhtP-QX7Qau97jxjtkY 4147
probjax/inference/kernels/slice.py sha256=SYE4HsKvzDRYAha-qLInGAFOuLPy6QVynY1pN3krSbU 8840
probjax/inference/rejection/univariate.py sha256=W7vZFqsN_ucQxD5byyhAI-0qeerdD8VOunLMdkwDj9o 7519
probjax/inference/smc/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/inference/smc/adaption.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/inference/smc/base.py sha256=vXvGNPjn_Tffj5R6A5cWp8vF1rq3f9REDoa6K-w0gms 948
probjax/inference/smc/resampling.py sha256=kWbtbTXKHN_dmNzMF3EyiJPKXN_CtCmyU4siXnobuF8 2909
probjax/inference/smc/tempered.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/nn/__init__.py sha256=11UorB0VnRIAax0CzPkZuLvsQJAD2BA_0_OjnmuEStA 680
probjax/nn/attention.py sha256=Fe0DWCXO8wmENgRGSnG1404DrLGWobUyhAgltVrurJE 9433
probjax/nn/bijective.py sha256=-wXloyvd0y3f0H1r0ZB-L4K3kc-LCtHADO7Z_uu6yeQ 12265
probjax/nn/utils.py sha256=Pcc7LopLq9EHoWhCzS70os4NkI8I6POr3OdMs6_iwYU 4754
probjax/nn/loss_fn/__init__.py sha256=fQFoIbMmAHrhOAq5y1v9S5PYA1IbAZHgca8xmot8foI 544
probjax/nn/loss_fn/denoising.py sha256=CybcwaIoIWEuNafYogROrUPjT7Cp9G5afeRIH2sGGJc 3151
probjax/nn/loss_fn/denoising_score_matching.py sha256=1823BneJ8YBhIH3vtLGstLhHUXuU6avaJ5WoP26bgiA 3478
probjax/nn/loss_fn/flow_matching.py sha256=At_AasbCrpQXrF9n7j9hhPkzntbKnV3DK-YcUb2iRwg 3797
probjax/nn/loss_fn/score_matching.py sha256=lKbDMB-nle2mlul49iPXgo0KM4JS_QKhNIylRZFZxjk 2745
probjax/nn/loss_fn/sliced_score_matching.py sha256=0UcEL0X3exOvmQjkR7qQYNKY_xOrSfzoNHPVCKHhPhw 3536
probjax/nn/loss_fn/utils.py sha256=cgL_NvSG_xFZhy_ASExsz2q5kxMKH_e4lvOYN4YYNuQ 245
probjax/nn/nets/__init__.py sha256=i9YBPioRjE_igdiEVYmyJjQDrtekH9FD9LuFJq8qi0E 289
probjax/nn/nets/autoregressive.py sha256=t5IE_tNCfX1gVER8o_Ew2-Cf5TdKcs3FoOVqeIT3wlQ 2405
probjax/nn/nets/coupling.py sha256=BfR42z1_hPMbQkzUHKDgw19_1Eam82iAL4Rbsg_nuDE 1205
probjax/nn/nets/deepsets.py sha256=u8vYs7nRmV4KOrcfv2iz92K_m8JKIW1w74nyMD-rzps 1926
probjax/nn/nets/diffusion_model.py sha256=SuOr1_8MDlcePaQWxUXD-vLS2OqZqiVMmQAQ8kTkhJw 4985
probjax/nn/nets/lru.py sha256=Hs-q-CwDNPOwsHclA3gUEh4PGB4Z3n1WNU_Znh0YLWk 8205
probjax/nn/nets/masked.py sha256=Y_KC6Wyqi0EWgedSXkIXO9LandrEtcW-Ovgs2MHrNVU 2647
probjax/nn/nets/mlp.py sha256=ADprmOErRVkjKtejkGKhRlCJGrXZIwWSlA1_engv2hI 1303
probjax/nn/nets/score_nets.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
probjax/nn/nets/transformer.py sha256=GMTkDzqJi4NS8Un0XQBFRadZnZ6-pAIEtT5xFiLH1TQ 9107
probjax/nn/nets/unets.py sha256=FKi_IBVJiFPvl0eTRILdS8jGSc90yxDqZNWfEpZW9OU 8665
probjax/utils/__init__.py sha256=uOMQg_1tpSjHr5JIAK6fRZDyhRkrrD1qdG5YG-PVS7A 14
probjax/utils/bmutil.py sha256=efS9ivHk14vYdV5p2vvYeBxu2urGdbeDRZYur3_bsf0 6432
probjax/utils/brownian.py sha256=X7xKW7m0eIzWikkYJkeChF9zJggxBB5rpWoNxqa111E 5794
probjax/utils/containers.py sha256=NWcxPNWDmOWURhCmbo4DOEOuTpjslLWRFmHb0RNdMHU 2324
probjax/utils/graph.py sha256=8iFeuAmaWSRznTmWpmRDIjBhuLRe3mCIo4jMZ6UFZto 7966
probjax/utils/interpolation.py sha256=OcDE3-g8RgLiX7hwNLwdkv2hkHYvZXi8Tl5uAVTNSRo 2534
probjax/utils/jaxutils.py sha256=kB9ZTrRBtfy46YmQ8BFUZzofyxe24s8_YQ69FfMRQSw 8311
probjax/utils/linalg.py sha256=EPIJSo0-eZFfERpyxdNAHHhMIbCOoaGTOhwaz__hal0 5011
probjax/utils/odeint.py sha256=BHXMJRGChDfRQiXmYuOkXYBQuDUmpmgXg2j7zUn2H4U 4537
probjax/utils/sdeint.py sha256=euWTV7_zkGP4kXOQ8-TemGXmYXJ3B3eAgERQXGjGx7s 4063
probjax/utils/solver.py sha256=f4_Zz8AKxn3gPtJ8Q_iCjyLRrttuA6lBmJoNob_Ll-w 3641
probjax/utils/stats.py sha256=HJ0nbfkQ5E6hTEorFAi5s3I8RgN6RHMvyv4xw8XGw2w 2812
probjax/utils/odeutil/__init__.py sha256=4cPEG9I6h2c9E_iLubOsYCQQyXuBqPkgIJ-_cd1GuM8 231
probjax/utils/odeutil/base.py sha256=shSavsCb0DD3EBPekXm12xXWyOq6rNQ2jt80Sr6_QK0 1709
probjax/utils/odeutil/exponential.py sha256=vJ4rcrRAq0Dsvsj2xAoqwuH06v1csSvWiXrtIw3Igjw 663
probjax/utils/odeutil/integrate_adaptive.py sha256=EpuTNK3uSdhRGzWNxX1lWO-1oLeNctnHzdznDY7QTyY 6792
probjax/utils/odeutil/integrate_on_grid.py sha256=J3a6ZcSsp6X7bQfJsuEJfgo_ZdYAQzsw0EDmr3wZ2uE 1375
probjax/utils/odeutil/rk_explicit.py sha256=rPhhf9BBiYBYK5idPUD05UnFgkkQkaZLE0JilnjjeuQ 19400
probjax/utils/odeutil/rk_implicit.py sha256=9lfOCwwsJTbupUbZS8hsaJ3XGH69pgpwwnaqJ57rm0Q 5067
probjax/utils/odeutil/util.py sha256=mi_tYPH7zCf0IJE4SO7_KXANZN6lpNfaq2pOBo9P_0c 3754
probjax/utils/sdeutil/__init__.py sha256=i1fjXeLyQ5nqYQ085icw1nQ7GxTEpbq-9ythZdht7l8 222
probjax/utils/sdeutil/base.py sha256=PmgRhUZpGH8hbtTfK3f3Pr7nFE_W97zc71KPTNYO8oE 1772
probjax/utils/sdeutil/em.py sha256=odiyZ_knZaYPO_v4rfb5Y8deT01pfpHtiA1ACko_JdI 2125
probjax/utils/sdeutil/integrate_on_grid.py sha256=TgmSOCdlL86ftBW7eyYYiYCGzU3ogd9WveM2dSQ4DmA 1594
probjax/utils/sdeutil/milstein.py sha256=lSH3DJHxrnZN3i4m0tYXpxfqxPy_v7DzS9_Q_HDDPyc 2108
probjax/utils/sdeutil/srk_explicit.py sha256=ZKZGONc2xKbUY7ZLUfGMtoSsUWSbqoATY2rnSNWRBFY 4480
probjax-0.1.0.dist-info/LICENSE.txt sha256=eaLzoEAL9didlDA6rDinooXDpmgNKR3X1_P2UVd9alE 1072
probjax-0.1.0.dist-info/METADATA sha256=VTKUOk3-mXDvMH2haJBAXEapum-NwMCSDRvb0s2xjyM 2715
probjax-0.1.0.dist-info/WHEEL sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc 91
probjax-0.1.0.dist-info/top_level.txt sha256=_IXDm9y_cKGDXhMruAsakzypf4V5shCt3aH7X5Li6gI 8
probjax-0.1.0.dist-info/RECORD — —

top_level.txt

probjax