dynamax

View on PyPIReverse Dependencies (1)

0.1.3 dynamax-0.1.3-py3-none-any.whl

Wheel Details

Project: dynamax
Version: 0.1.3
Filename: dynamax-0.1.3-py3-none-any.whl
Download: [link]
Size: 144915
MD5: a081e2923452e57de064ef7ef3fdca19
SHA256: cc83fb67a03ce7ac39beaf2e8528a550763aacd050524ac099757abd11b4f3d3
Uploaded: 2024-02-27 23:17:09 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: dynamax
Version: 0.1.3
Summary: Dynamic State Space Models in JAX.
Author: Peter Chang, Giles Harper-Donnelly, Aleyna Kara, Xinglong Li, Scott Linderman, Kevin Murphy
Home-Page: https://github.com/probml/dynamax
License: MIT
Classifier: Programming Language :: Python
Requires-Python: >=3.6
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: 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: 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: 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: codecov; extra == "test"
Requires-Dist: coverage; extra == "test"
Requires-Dist: pytest (>=3.9); extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Provides-Extra: dev
Provides-Extra: doc
Provides-Extra: notebooks
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 6855 characters]

WHEEL

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

RECORD

Path Digest Size
dynamax/__init__.py sha256=JJ3px78rQnFEd2UoxqHlEq82bujL9X1G2w3dm7WCPKI 265
dynamax/_version.py sha256=3UQCm54Y0SNILwvz6vWDYrk8FltxHTz5G_2lQuO5Fe0 497
dynamax/parameters.py sha256=H-49ExVtg-xKHxm002OL7sXdVhuJBXOwtfeFqLLdrGA 4897
dynamax/parameters_test.py sha256=fsxL1aRVa1mtKH5WllGGinnWr59znrTMOEP5JkPkkbg 5304
dynamax/ssm.py sha256=ya9YTtjcuX1oyLLx7en3Fnd79Mo-mIJhfO4tuTSmFZo 18732
dynamax/types.py sha256=wRuh2rdDJvjA4VY_v6nQCjdvPg-LIqq-MNU67J4amZk 266
dynamax/warnings.py sha256=PBW3nDAPT9MjJ7enG0Ip580s4mxeQdggjN1X_cgv3co 621
dynamax/generalized_gaussian_ssm/__init__.py sha256=Iuat6gp9EjyIoUumdfS4IcoHnijy7Ks_PSP1fq5bMW4 570
dynamax/generalized_gaussian_ssm/inference.py sha256=RmaHaxxj_H7wCXj6AnLlCdlI5lFnQ9HnmpEYGnarXkQ 15343
dynamax/generalized_gaussian_ssm/inference_test.py sha256=7v7oCoNyH4TTwRAgpROMyM3EsQ05zestWrW58M9p8D8 3442
dynamax/generalized_gaussian_ssm/models.py sha256=_kP7I88Ai5scrYp5GSffwnSf_KiXW8rP8oUUpeyWGIs 4937
dynamax/generalized_gaussian_ssm/models_test.py sha256=NuRDk0XGqvCWjgQtWXK0JyopQJXDfjQcCZuxT_f0bUE 4487
dynamax/generalized_gaussian_ssm/demos/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/generalized_gaussian_ssm/demos/cmgf_logreg_estimator.py sha256=LugHb7-HLaTuxVhKj4gd82VG_RADmz-dauU5kKbz6NA 2554
dynamax/hidden_markov_model/__init__.py sha256=uwEgjmlD51wpkcjCKPEf4xFQ5Gd7tVYIqBI9u8--McI 1984
dynamax/hidden_markov_model/inference.py sha256=6Xmi-DtnnWZpwcr06alchrTMpAq_Lnz4Vzox2jRFbjQ 26279
dynamax/hidden_markov_model/inference_test.py sha256=Kk0BudA2W-jevYh57b5fYlvwDH5YS40TxEwbSZ9wdKc 11953
dynamax/hidden_markov_model/parallel_inference.py sha256=pzb8TTIA_ISKWOBJaGBsmcqvKmwlY4E4xrKw1_2EU2w 4312
dynamax/hidden_markov_model/models/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/hidden_markov_model/models/abstractions.py sha256=QgKuaT8nFce0Gqp4mpv2pGbuomZdAUYtTWmEcLItaa8 29011
dynamax/hidden_markov_model/models/arhmm.py sha256=dFMLuGXOgj0KRqIlyHS4gs2nWktoNg2DaKvncMuoiXg 12422
dynamax/hidden_markov_model/models/bernoulli_hmm.py sha256=-Wv8u7ydt_UeXdvoT6YxYm24VCtd_BCa8sRr9aEHkzM 8510
dynamax/hidden_markov_model/models/categorical_glm_hmm.py sha256=RJNZ6jh3KUuGoXhTAj62PxZz7SZI0h1H21eKD_DEVac 8696
dynamax/hidden_markov_model/models/categorical_hmm.py sha256=mOhjt1fPqksMoR6TRIOGflCI-o5UmsKlRzS3aCMEceM 9222
dynamax/hidden_markov_model/models/gamma_hmm.py sha256=_PnAVJvKLChj6YVzYJZ-6tKqmmjSrqRxw6YmaT2Hu5Y 7297
dynamax/hidden_markov_model/models/gaussian_hmm.py sha256=lFRJTYQ_ITeTbS6jFUfEwhJiP8kIiOqnzvJtvj_KFlo 55329
dynamax/hidden_markov_model/models/gmm_hmm.py sha256=PP50XLHoJT1_f-WifhLxthrnOaLHAg2BZpqIjU8lNwg 27862
dynamax/hidden_markov_model/models/initial.py sha256=xovMEgqODHrH6k7ynFO71ibwwEBGxL1Pjw_OBhX1HuQ 2803
dynamax/hidden_markov_model/models/linreg_hmm.py sha256=iIVInllzR8wkTEOiE8rjfLF8nDS98bpKZfhnT37vLuY 11209
dynamax/hidden_markov_model/models/logreg_hmm.py sha256=IqTst6Cu5ULNDdFEGybYU_DbHYq4M0Y_VOROV-Z7V0o 8968
dynamax/hidden_markov_model/models/multinomial_hmm.py sha256=TcxZWw3Jq40UVlzU8QOczYwsoSU5IJKNI3-E923xERo 8223
dynamax/hidden_markov_model/models/poisson_hmm.py sha256=xZ4pWkDUy4iDQSI_Y5vOiQASxopx6bHAln503vTjzrs 8254
dynamax/hidden_markov_model/models/test_models.py sha256=7La-h-2d8l9i6AmpLImy-IoceaoAJg7TuH4mnu86AyQ 8570
dynamax/hidden_markov_model/models/transitions.py sha256=tCVhJUkbCPuHt72hQVSevTCFTp1jJUFdV5K8_GMj1EM 3291
dynamax/linear_gaussian_ssm/__init__.py sha256=NNtq9nTzk8zq39-f6fqy8CH0lZ9YeF-6pfBi2yzTk3I 1478
dynamax/linear_gaussian_ssm/inference.py sha256=sF1-UDUTYxD3ltl4pDNrEFi7okS4E01XW49C_-qwY0c 22710
dynamax/linear_gaussian_ssm/inference_test.py sha256=uoXkRhfMGyNdt3O4gnAzuVeESpVnhmWNuLUYrK4VOSk 9676
dynamax/linear_gaussian_ssm/info_inference.py sha256=aXvsz-3spDVWn4AGHiGwMK7JSjIlpAaThL9mYqTnpjU 16697
dynamax/linear_gaussian_ssm/info_inference_test.py sha256=7EKG_1Bk8uYHeEsUgiLxLC3TWVe2pvLEdY_QhhoSaoc 6313
dynamax/linear_gaussian_ssm/models.py sha256=5KSx324Db_tOZnK9ifilKmbkQr9JxK4qCUz6FuHXd_0 25826
dynamax/linear_gaussian_ssm/models_test.py sha256=1wObElq4ecSkT9OCVm39PB72YDRh_nWZSFlBBTnZz-A 1061
dynamax/linear_gaussian_ssm/parallel_inference.py sha256=RqWjWu06AHeIiI7AIo9DqKQYacxQybYBBaOCQb_PoE4 13565
dynamax/linear_gaussian_ssm/parallel_inference_test.py sha256=vdrWakier47m3QRNiCMGgLWLP1oK3kq5cqNZ-U2VkOE 8572
dynamax/linear_gaussian_ssm/demos/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/linear_gaussian_ssm/demos/kf_linreg_jax_vs_pt.py sha256=mBUtazMWH_vkyqSpFXPNRsdxBqQye4HSEtGGC8EGiaw 10260
dynamax/nonlinear_gaussian_ssm/__init__.py sha256=KtgR7eAmuVaqF_w30ULFhp6dkuMR64hfSqPVDlxtLWQ 752
dynamax/nonlinear_gaussian_ssm/inference_ekf.py sha256=kyg6G8NYXSYFbmEB-do4ZAJCb-A63xzdtnkxy9RSR1I 12236
dynamax/nonlinear_gaussian_ssm/inference_ekf_test.py sha256=lIriMb_RNo_CkkIR4DC4j03-aM7fbeuESz2YpCgQwZc 4400
dynamax/nonlinear_gaussian_ssm/inference_test_utils.py sha256=ITa20L689woVrigidESfVeHDbI7ZMs9W8US8ttVziGk 7426
dynamax/nonlinear_gaussian_ssm/inference_ukf.py sha256=BRVH4k7sNDZBj0OTHN96dGEzhVHOt87OjkHLQEeGZ3c 10751
dynamax/nonlinear_gaussian_ssm/inference_ukf_test.py sha256=7_KPEiC7fwIK3LVWhJXHo2IQ9gJcslWwkxQ1z5rxG_M 1305
dynamax/nonlinear_gaussian_ssm/models.py sha256=FqTpu6WKl-0VgMvaPeSI8L5yowFA1oJEuEWEL-jFKYg 3989
dynamax/nonlinear_gaussian_ssm/sarkka_lib.py sha256=KqDmUkSEhNCoLg2wvZ9xvec6dAFW07tMD-zte64DX4A 7209
dynamax/slds/__init__.py sha256=YooCuEd9MfU3Pm9og-fJLGsRcBsgu7d_s5V3BYC1z3I 147
dynamax/slds/inference.py sha256=XAH1BG8QDnSGj2tVBjZzT1dWiOPZunTvXSFPlJ0rtcM 16233
dynamax/slds/inference_test.py sha256=kEuWTCtpG5FrGsxTWfvP4yifzv_kG03ftXLk8hAUbtQ 4412
dynamax/slds/mixture_kalman_filter_demo.py sha256=ceqK3sV-7Dyb2DFEb9wJaBxgfHghI8Brb3jKzITDcQE 5311
dynamax/slds/models.py sha256=AoQVeqrBLd90GJbZKBvwlFZRypBkBNxB3LqHzRtmm-4 5931
dynamax/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dynamax/utils/bijectors.py sha256=72wKnGA7VziBGOkniHOfQiY04uy1-9_vYnCbYwaA7UM 1111
dynamax/utils/distributions.py sha256=BYEEQczuqujjbPNrV57j2Xn30_RAgQzzOZWLE0s_Cxc 15052
dynamax/utils/distributions_test.py sha256=JhD5GfBE74JsfOdg5mosgw2b-oAo_lLBNB0rM8iQD70 6865
dynamax/utils/optimize.py sha256=JMU9rRQ7BBN7kRy3dfW0e27wyT2zxGVJ7vYrie4xfbU 4119
dynamax/utils/plotting.py sha256=Jlz0yG31Y_0yMWcTh6BVi0rLISsA0C5v_UnzJNKcgJE 5169
dynamax/utils/utils.py sha256=HV9N5SFanhccmfzUNXpXNfKCRCl28U70DtpiIh-7f3w 6477
dynamax/utils/utils_test.py sha256=uBAxrfeUsm-EE89mZWXlNrQ1xd32B23iUuIprsZrl90 1507
dynamax-0.1.3.dist-info/LICENSE sha256=Zml6D-b6HsuCm5jrtPrHP3h3GERyDcdztCsFuMvrheo 1087
dynamax-0.1.3.dist-info/METADATA sha256=UriPaBbaaTxQveoTrUMt3qtHRUUmTtQZyNSFOxkRus0 9319
dynamax-0.1.3.dist-info/WHEEL sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM 92
dynamax-0.1.3.dist-info/top_level.txt sha256=FOkfS-fTd_ULBmJxF_pmrK6Eb8CFZBtjRSga_pIANsE 8
dynamax-0.1.3.dist-info/RECORD

top_level.txt

dynamax