dynamax

View on PyPIReverse Dependencies (1)

0.1.9 dynamax-0.1.9-py3-none-any.whl

Wheel Details

Project: dynamax
Version: 0.1.9
Filename: dynamax-0.1.9-py3-none-any.whl
Download: [link]
Size: 155256
MD5: 27c62186f9833e12d086d2a4641160c8
SHA256: 8c5acaa9ab0867abf7910cb096c619c7976fa804172fe482bc60f631be576bfd
Uploaded: 2025-03-07 16:49:01 +0000

dist-info

METADATA

Metadata-Version: 2.2
Name: dynamax
Version: 0.1.9
Summary: Dynamic State Space Models in JAX.
Author: Scott Linderman, Peter Chang, Giles Harper-Donnelly, Aleyna Kara, Xinglong Li, Kevin Murphy
Maintainer-Email: Scott Linderman <scott.linderman[at]stanford.edu>
Project-Url: homepage, https://github.com/probml/dynamax
Project-Url: documentation, https://probml.github.io/dynamax/
Project-Url: repository, https://github.com/probml/dynamax
License: MIT License Copyright (c) 2022 Probabilistic machine learning 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.
Classifier: Programming Language :: Python
Requires-Python: >=3.10
Requires-Dist: jax (>=0.3.15)
Requires-Dist: jaxlib
Requires-Dist: fastprogress
Requires-Dist: optax
Requires-Dist: tensorflow_probability
Requires-Dist: scikit-learn
Requires-Dist: jaxtyping
Requires-Dist: typing-extensions
Requires-Dist: numpy (<2.0)
Requires-Dist: matplotlib; extra == "notebooks"
Requires-Dist: seaborn; extra == "notebooks"
Requires-Dist: flax; extra == "notebooks"
Requires-Dist: blackjax; extra == "notebooks"
Requires-Dist: graphviz; extra == "notebooks"
Requires-Dist: scipy; extra == "notebooks"
Requires-Dist: matplotlib; extra == "doc"
Requires-Dist: seaborn; extra == "doc"
Requires-Dist: flax; extra == "doc"
Requires-Dist: blackjax; extra == "doc"
Requires-Dist: graphviz; extra == "doc"
Requires-Dist: scipy; extra == "doc"
Requires-Dist: sphinx; extra == "doc"
Requires-Dist: sphinx-autobuild; extra == "doc"
Requires-Dist: sphinx_autodoc_typehints; extra == "doc"
Requires-Dist: sphinx-math-dollar; extra == "doc"
Requires-Dist: myst-nb; extra == "doc"
Requires-Dist: jupytext; extra == "doc"
Requires-Dist: sphinx-book-theme; extra == "doc"
Requires-Dist: codecov; extra == "test"
Requires-Dist: coverage; extra == "test"
Requires-Dist: pytest (>=3.9); extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: interrogate (>=1.5.0); extra == "test"
Requires-Dist: matplotlib; extra == "dev"
Requires-Dist: seaborn; extra == "dev"
Requires-Dist: flax; extra == "dev"
Requires-Dist: blackjax; extra == "dev"
Requires-Dist: graphviz; extra == "dev"
Requires-Dist: scipy; extra == "dev"
Requires-Dist: sphinx; extra == "dev"
Requires-Dist: sphinx-autobuild; extra == "dev"
Requires-Dist: sphinx_autodoc_typehints; extra == "dev"
Requires-Dist: sphinx-math-dollar; extra == "dev"
Requires-Dist: myst-nb; extra == "dev"
Requires-Dist: jupytext; extra == "dev"
Requires-Dist: sphinx-book-theme; extra == "dev"
Requires-Dist: codecov; extra == "dev"
Requires-Dist: coverage; extra == "dev"
Requires-Dist: pytest (>=3.9); extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: interrogate (>=1.5.0); extra == "dev"
Provides-Extra: notebooks
Provides-Extra: doc
Provides-Extra: test
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 7290 characters]

WHEEL

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

RECORD

Path Digest Size
dynamax/__init__.py sha256=vVdR-OwEs7dWk_WOZsDKRaP_PysafOTLS2_snJT7vbI 338
dynamax/_version.py sha256=zT27FvLt_a_Enx9e15wKdTk3A0F9PuNQkFOEhZApwRk 497
dynamax/parameters.py sha256=2IcWLXo1v4nVkq5zSWIeJttsiI35GU09iVsQXPzNegg 5442
dynamax/parameters_test.py sha256=66idpSbu19YVoJzLWLaLz-SoTle87WYbwaZe_nwbd-M 5825
dynamax/ssm.py sha256=W_0tm3frJ91nvjqA0U7MY55fUXhd8rxytI55SXsA2A0 19047
dynamax/types.py sha256=9oltnbA7ZLJAJsNWer0bh3nDfQqP2Rhz5hC-TUuaIPE 289
dynamax/warnings.py sha256=UXz4bRkpRO2RlL1a_4hSclM2HD5WHkgB2zPkTYO1SCk 690
dynamax/generalized_gaussian_ssm/__init__.py sha256=ZqJX0h3TN-wZB0DV8Gog-2xhJ_PjSLSjJ8-MTavazos 369
dynamax/generalized_gaussian_ssm/inference.py sha256=W_jkDfdebCsh6SVRFchwj9ogZi5B4zCmjfkMDMC7LHY 18786
dynamax/generalized_gaussian_ssm/inference_test.py sha256=J280OQNehjmS7qU2K9Tivlh4WpQpQBwSaG9faIWiZzc 3259
dynamax/generalized_gaussian_ssm/models.py sha256=GHbCLIrmq3yhO4mlupL1j1K3GU2q7i_kmU6JJhzIpYU 5493
dynamax/generalized_gaussian_ssm/models_test.py sha256=gocdn3bDarZEUMs6Y9PtvnBFAolKn2RqR-DwZG_rqOE 2605
dynamax/generalized_gaussian_ssm/demos/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/generalized_gaussian_ssm/demos/cmgf_logreg_estimator.py sha256=Y_tZ8l9uljkHS6Y2EW1yiq_Kkd2SMEIaI7uXiD8ABeQ 3166
dynamax/hidden_markov_model/__init__.py sha256=0zeshbM9VLijyoZgRQ7h2zfUB5ofQoyS9P8kqdi5SwQ 2097
dynamax/hidden_markov_model/inference.py sha256=Lgmv4TgPnFc8p4xY5F_P_IjD9-jMzFLHdUodvPyqxA4 28014
dynamax/hidden_markov_model/inference_test.py sha256=Z5Lpt3rgm82uuGM02chbwN-ZqbBl3QylaLHfUXDwkZU 13820
dynamax/hidden_markov_model/parallel_inference.py sha256=VyAZgYyB96pRROMP7GaBnKgunn-4rotys29WZ40komM 8156
dynamax/hidden_markov_model/demos/categorical_glm_hmm_demo.py sha256=ecfy2qXwsSgk5OsXcHGKKjbyxAfs5s-8-uYKhRf3j7k 2549
dynamax/hidden_markov_model/demos/poisson_hmm_earthquakes.py sha256=mL2zhio8tFc6GjJmGs2yGFsxgZC9qtZBZHG_7hWkqQc 3622
dynamax/hidden_markov_model/models/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/hidden_markov_model/models/abstractions.py sha256=YbyalHi_o0MdL6cnz_EqmKtAqbzrzngy63TuiATNNaA 27894
dynamax/hidden_markov_model/models/arhmm.py sha256=GE8-96ZJ-_MHkEo1GU8HCtfw-rvJGDYhZTiEH1k7fWA 13768
dynamax/hidden_markov_model/models/bernoulli_hmm.py sha256=LuQboXBZSYosnoUJlh2kEKzOfbX6FmIW0UD_l-ukoqQ 9936
dynamax/hidden_markov_model/models/categorical_glm_hmm.py sha256=mN7NMn2xrFOqgvkl5U8IB37jvFId9S0lCM0nFUJc8SA 9716
dynamax/hidden_markov_model/models/categorical_hmm.py sha256=itiHsQtEfSgnXv8Sltkg8xyrYPEOzYZgLWh_1Xvl6DE 10246
dynamax/hidden_markov_model/models/gamma_hmm.py sha256=FgBve25H_VlOxNjR1cquhzU8UgWFadDXZbLkH_x64zk 9076
dynamax/hidden_markov_model/models/gaussian_hmm.py sha256=8KHsZtBvY-1GXFI6iHZ-QkfCusKk0A3VGN_Kn7sjCX4 65732
dynamax/hidden_markov_model/models/gmm_hmm.py sha256=Qc_gguS0OP08vznaIVBW5o6d7hHaGMk_eubEKStjuDE 34700
dynamax/hidden_markov_model/models/initial.py sha256=eIbVuZtoXPC5s_nRS9kdk5y3VlhxQC49aVhhxIahV3c 4294
dynamax/hidden_markov_model/models/linreg_hmm.py sha256=v8wCXaxFWXawkjQQuqEnhIxvR48OYw9BgtrUbwFtKeU 13816
dynamax/hidden_markov_model/models/logreg_hmm.py sha256=Og2VuW4uiLlTHCGDKlOCOHvZQ0lTZYWjK6XQgBsmHn0 11286
dynamax/hidden_markov_model/models/multinomial_hmm.py sha256=k0LT5PvZUDAKpBNPGJactlVZT9JBTFM9T1_oyi8ZYgQ 10651
dynamax/hidden_markov_model/models/poisson_hmm.py sha256=BBhHgUCeviaqpV4ryCcTpNbGvx75W34RB1Iq6dYFkZE 10303
dynamax/hidden_markov_model/models/test_models.py sha256=QlJ3eIx2kmQ0cQrSE-2Pi7Qgl9RQpHwHbV6StoDdNtI 8811
dynamax/hidden_markov_model/models/transitions.py sha256=kcewR8ARbaFCKkhqSu5z-sXNeGNdwHlc13UQf6oDT_E 5028
dynamax/linear_gaussian_ssm/__init__.py sha256=NNtq9nTzk8zq39-f6fqy8CH0lZ9YeF-6pfBi2yzTk3I 1478
dynamax/linear_gaussian_ssm/inference.py sha256=dw74ZUPpfN8vjfnbXZupQnVcB9wThbdIRWb1l_V2RZc 25502
dynamax/linear_gaussian_ssm/inference_test.py sha256=inp9mykzQYS_6oBbzXGCBj2P88GW5QW4340v50km50k 8397
dynamax/linear_gaussian_ssm/info_inference.py sha256=bhz0AVv98xPzb8cNxXaTZouXDL7XyVFJYGLmWJ1tOA0 20045
dynamax/linear_gaussian_ssm/info_inference_test.py sha256=YqkecpLUlgf-s-0T6dnHP3xYDQ9vMOoMHY890IiJY70 6586
dynamax/linear_gaussian_ssm/models.py sha256=QLbposmp1hcWwnMdnrMCgYfPUyGLCvoO4Me0R8o0E3E 33501
dynamax/linear_gaussian_ssm/models_test.py sha256=myhP2r0jwcZaG6Kk7ujUasyXowfI7ovT3_kuqXrAOew 1152
dynamax/linear_gaussian_ssm/parallel_inference.py sha256=idA6umflqoVS5YXPW392PX3Bfjjnzx2lJznzb7ezX4Q 15690
dynamax/linear_gaussian_ssm/parallel_inference_test.py sha256=qyUMNXtRjRWS3uYeFvTi6CUvRUEwu5i57XxhZ8l6xZc 13226
dynamax/linear_gaussian_ssm/demos/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/nonlinear_gaussian_ssm/__init__.py sha256=oplwqlqq8J4CJO9tsyB4FqdDiLUIfxIrWZO5slszPlU 572
dynamax/nonlinear_gaussian_ssm/inference_ekf.py sha256=ewcSOfbSxuoya2zwvf0xrV52_SsRJoCNMO-5t1peM6c 11755
dynamax/nonlinear_gaussian_ssm/inference_ekf_test.py sha256=WLUy5BR6192ON7jxd8ufA-15JJ2HK2BigoonWbOAPAc 5105
dynamax/nonlinear_gaussian_ssm/inference_test_utils.py sha256=vaTC109p9A7U4TqTHmwr4hdAF9qRv5HV6En-TefbvK8 7681
dynamax/nonlinear_gaussian_ssm/inference_ukf.py sha256=iIO7ggtPpnbOeO91w9G7sVxe6K2pyioNjm55ApylSuY 12603
dynamax/nonlinear_gaussian_ssm/inference_ukf_test.py sha256=FgZX_wEH1IVO9G5g0FqHOjVZXPb9MSkY3eYV4NB6Rsg 1430
dynamax/nonlinear_gaussian_ssm/models.py sha256=rc6_8NAWVWQCSAKUgHyVZsQRzUkuSoAy-R4ypG6wLBs 4523
dynamax/nonlinear_gaussian_ssm/sarkka_lib.py sha256=_wnSsJ_pLGQKVyl0W6NVkBP9ALNtEojdb2zb0kb4vBA 7553
dynamax/slds/__init__.py sha256=YooCuEd9MfU3Pm9og-fJLGsRcBsgu7d_s5V3BYC1z3I 147
dynamax/slds/inference.py sha256=dCK736iUx34o2hOoT9BviLAANcKPz_8clFjkv_YdiqQ 16425
dynamax/slds/inference_test.py sha256=73M3dYAY2dNYcEsadSNkeaEntvZfwp1vnV643PKyY1g 4581
dynamax/slds/mixture_kalman_filter_demo.py sha256=1EFVltVxI03cpfr12zrb0LAj1yBe_mhCwzOWMy30QaY 5691
dynamax/slds/models.py sha256=nquBDM9l0QlGFNZoEfNnsiPAhHLTlX-CloK-WNhjPNo 6218
dynamax/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/utils/bijectors.py sha256=p-1EFHWqbWdpJIMcRxLWazkaUL-ijqJHBlCgSZ4T_qE 1368
dynamax/utils/distributions.py sha256=RALnfqoQzpzVjlLVWK22U4dUPU2V4rSaJ8YcZO2Md5g 16296
dynamax/utils/distributions_test.py sha256=9eVHpwMkP5LgMdY4dPWLEVvwcmqB3XpXRs26f2HZ4tQ 7580
dynamax/utils/optimize.py sha256=MIT25wMcN0fwF-7qE_5HcTu48wycNPinltDo1eMPNYE 4432
dynamax/utils/plotting.py sha256=GS6jeQMeDAahq9YW14QPIOm8ShzUDc8BSCSp5u-2re4 5305
dynamax/utils/utils.py sha256=B48ZeVkEV86fN6A20vholbc8sT6vp5pUlrpvN9zFLB8 6894
dynamax/utils/utils_test.py sha256=ecU7EGVSMrKsEArAtJgBPMk63mynTRfkOsnFsi48x8c 1547
dynamax-0.1.9.dist-info/LICENSE sha256=Zml6D-b6HsuCm5jrtPrHP3h3GERyDcdztCsFuMvrheo 1087
dynamax-0.1.9.dist-info/METADATA sha256=qPvMc9BETcMz1AWgQn6GlWJ3CcpUUxlLoxXIoIDW6Qs 11270
dynamax-0.1.9.dist-info/WHEEL sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM 91
dynamax-0.1.9.dist-info/top_level.txt sha256=FOkfS-fTd_ULBmJxF_pmrK6Eb8CFZBtjRSga_pIANsE 8
dynamax-0.1.9.dist-info/RECORD

top_level.txt

dynamax