coax

View on PyPIReverse Dependencies (0)

0.1.13 coax-0.1.13-py3-none-any.whl

Wheel Details

Project: coax
Version: 0.1.13
Filename: coax-0.1.13-py3-none-any.whl
Download: [link]
Size: 187211
MD5: d6c839f6ba2a48b2981bf061407bf494
SHA256: 6f1b6e326603a1170e4f33b34a0f6f89980e4f41db37ba1b31a526c896ec3161
Uploaded: 2023-02-01 18:51:10 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: coax
Version: 0.1.13
Summary: Plug-n-play reinforcement learning with Gymnasium and JAX
Author: Kristian Holsheimer
Author-Email: kristian.holsheimer[at]gmail.com
Home-Page: https://coax.readthedocs.io
License: MIT
Classifier: Development Status :: 1 - Planning
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Framework :: Flake8
Classifier: Framework :: IPython
Classifier: Framework :: Jupyter
Classifier: Framework :: Pytest
Classifier: Framework :: Sphinx
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Natural Language :: English
Classifier: Operating System :: MacOS :: MacOS X
Classifier: Operating System :: POSIX :: Linux
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: ~=3.6
Requires-Dist: Pillow (>=7.1.2)
Requires-Dist: gymnasium[accept-rom-license,atari,box2d] (>=0.26.0)
Requires-Dist: numpy (>=1.21.6)
Requires-Dist: scipy (>=1.7.3)
Requires-Dist: pandas (>=1.3.5)
Requires-Dist: dm-haiku (>=0.0.8)
Requires-Dist: chex (>=0.1.5)
Requires-Dist: optax (>=0.1.3)
Requires-Dist: tensorboard (>=2.9.1)
Requires-Dist: tensorboardX (>=2.5.1)
Requires-Dist: lz4 (>=4.0.1)
Requires-Dist: cloudpickle (>=2.2.0)
Requires-Dist: importlib-metadata (<5.0.0)
Requires-Dist: dm-control (>=1.0.7)
Requires-Dist: flake8 (>=3.8.4); extra == "dev"
Requires-Dist: pylint (>=2.6.0); extra == "dev"
Requires-Dist: pur (>=5.3.0); extra == "dev"
Requires-Dist: pytest (>=6.1.2); extra == "dev"
Requires-Dist: pytest-cov (>=2.10.1); extra == "dev"
Requires-Dist: pytest-xdist (>=2.1.0); extra == "dev"
Requires-Dist: twine (>=3.2.0); extra == "dev"
Requires-Dist: jupyter (>=1.0.0); extra == "dev"
Requires-Dist: sphinx-autobuild (>=2020.9.1); extra == "dev"
Requires-Dist: jax (>=0.3.25); extra == "doc"
Requires-Dist: jaxlib (>=0.3.25); extra == "doc"
Requires-Dist: Sphinx (>=3.5.4); extra == "doc"
Requires-Dist: sphinx-rtd-theme (>=0.5.0); extra == "doc"
Requires-Dist: nbsphinx (>=0.8.0); extra == "doc"
Requires-Dist: sphinx-tabs (>=1.3.0); extra == "doc"
Requires-Dist: sphinx-copybutton (>=0.3.1); extra == "doc"
Requires-Dist: sphinx-notfound-page (>=0.5); extra == "doc"
Requires-Dist: ray (>1.9.0); extra == "ray"
Provides-Extra: dev
Provides-Extra: doc
Provides-Extra: ray
[Description omitted; length: 1830 characters]

WHEEL

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

RECORD

Path Digest Size
coax/__init__.py sha256=3UuAUm_Y2VPC4tAFttBSvMQnEfXwDYwFN_HQVOmM4jk 2853
coax/typing.py sha256=dkLcxNQE5fz7d_lGoRrpsInWW8f8cWks4FDmo_IfrFg 573
coax/_base/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
coax/_base/errors.py sha256=HNNnZCGHdNyV50F7pNcbOdCTHKA0fu_iqF3I1N0e2S0 662
coax/_base/test_case.py sha256=EfsHz1HGfSHgkt2JOBw74X1YiWXwN1a3o25hgfQ2Co0 13055
coax/_base/mixins/__init__.py sha256=TJbE5onhCY0pyCICbLdH1bRe43nSp_CcLVZw-N6sAQU 266
coax/_base/mixins/_add_orig_to_info.py sha256=Y0Iu2oJnYb45TeVEh4bY4byyj8mnQj7CvD6AuztMb6M 895
coax/_base/mixins/_copy.py sha256=QEYVcv82wxmi1ofgOnlxWeJQbdPpEWWvlAe7tnkHkTQ 429
coax/_base/mixins/_logger.py sha256=hmQYpJhzs4VRKiiKCQqXnTsvDethJs29klZqT5KnQoE 130
coax/_base/mixins/_random_state.py sha256=3pErygFNkTD3q0FxRx5cdq5z234dI0Z4DLS4A1eFHho 526
coax/_core/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
coax/_core/base_func.py sha256=9RXGKkK3cW-Rl-Zo3l2rxVnyKttr1u48wY9K5oKfMNU 5555
coax/_core/base_stochastic_func_type1.py sha256=yM2HaqBO5vN8fNjJlUx3Z1Q3OiCtZPgN--46M8p5fk8 19163
coax/_core/base_stochastic_func_type2.py sha256=oQzjw9aSiusHjLcVaBfDjhEgMNx7M7iUdhDKV2olgM4 7645
coax/_core/policy.py sha256=rGlCcXrcXmcWDYdURROnKPGbIXUe9yXAXsb8hcnPG-w 4373
coax/_core/policy_test.py sha256=P8ZZOT0kWJ6f_UbnoLaotxmX0JVNHtyx8U0nA_nL8pw 5038
coax/_core/q.py sha256=B7PQc0hB3SLMvf-O6GfNFEKlbi-uKI3SBc7jZ9V_e7U 10806
coax/_core/q_test.py sha256=3FyiYCHbrxhBwHaunfG6hKu471gSSC-LijQNSqNf9SY 7736
coax/_core/random_policy.py sha256=GMFwXyJkSHmJccLLU1Gsddfii8XmatAG4PW3T9AwCPE 2336
coax/_core/random_policy_test.py sha256=CTEGRyVf37Kl3nmoJs53X6oGTlix1WRnCFQSfBQvpuU 1627
coax/_core/reward_function.py sha256=YrlVt1gBhmRk4OoRPI5LSLxv3BZEdZd1gViTs_pd1ug 1889
coax/_core/stochastic_q.py sha256=2frUPJAAmetRcImSBSjy-dUcFKIfRlBQAWsLdZfZiQ8 9039
coax/_core/stochastic_q_test.py sha256=QUyUgLGKoXKilCLuWbKvIMREi2NfK8dtMLSCPVqRukQ 14291
coax/_core/stochastic_reward_function.py sha256=MNTP0H90LWyBdj3h5m6oC-a425K2k_3Bt0dBXbQgS9c 2642
coax/_core/stochastic_transition_model.py sha256=zrFiRhxRyFOdFj3nshTFL2LPYxaqwli1HFlzEgxFaNA 6627
coax/_core/stochastic_transition_model_test.py sha256=e4HCBn8-aGw-_OJdPYY3xG4ZmgND1JUxwVydpXReYus 14579
coax/_core/stochastic_v.py sha256=SVXcSx51FNkHsyyPMIbTYH_f1YqAizdQEqhcfrw9y1U 6909
coax/_core/stochastic_v_test.py sha256=B4TWcANjc5ymG3vWwNRYydWHmcDEowfpnpguyVesdKc 4708
coax/_core/successor_state_q.py sha256=uxfdKtPeP46XIdHEstkPlh68bYLOhb3LvAVnkkVOiTk 9699
coax/_core/transition_model.py sha256=HRmzhU-RlxzZIzx8iS1OsDCvT7p5Wh2DEBxsFG9hV38 12966
coax/_core/transition_model_test.py sha256=lyORvY9DRjsVQTo_zM8Rey7pr3QOBfbJYKEdU2oPBE0 8653
coax/_core/v.py sha256=hJhqE9jP8ai7ugFglLbos0JKY43_MCkSpAoWVMsbp3w 5528
coax/_core/v_test.py sha256=k6wiMs4_3lDN6zHgZ7BymGLaTNRWclIdZqoSEKaIXo8 3296
coax/_core/value_based_policy.py sha256=kIeHESGYdedtkvbzRRPkU057X_xG-TrOhdDnjohP5Ro 7759
coax/_core/value_based_policy_test.py sha256=1L5Z4SAhb-W78VssO7Pb6PQp4aCRnpQ88iAGo8KPVEA 3106
coax/_core/worker.py sha256=BKfv5aW88uorXY9EKHNn-_TR_tkfmNs0-tIl8vJrOoE 11753
coax/envs/__init__.py sha256=5J8wt3buJx6KuDkP8XrD8jWdh98Ci_gIlFgG1Zx9Xio 377
coax/envs/_connect_four.py sha256=aytL7o0-Ns-PR5ft0Cwj33oHTDbVjGpKBz3U8zyZzVY 10349
coax/experience_replay/__init__.py sha256=SJ138AZM7ouruX0QYB4STkL9F4k_IN1yBI2whz75tcM 785
coax/experience_replay/_base.py sha256=RLCo9RWK_8v7Ab8yMKgzSVt62uPnHBNetZQUfr-Q0yE 549
coax/experience_replay/_prioritized.py sha256=JkeN3v_mKBHKJTgOSgi13s0b777DKrAgMtf-NZ_ShfE 7835
coax/experience_replay/_prioritized_test.py sha256=y8YYfJY0y7Fp3DUbgeEznesBJmRdSty-iN8MnasNruM 4654
coax/experience_replay/_simple.py sha256=eOmC09hJvlsMEc-ZBgmfLLMBHUVH2NqnSBjktBT-yD8 2588
coax/model_updaters/__init__.py sha256=4Ugj7cs8ftJj2V4JJnAGFvLX6aO4__kJzfqkMgX4HoY 472
coax/model_updaters/_model_updater.py sha256=V_038wGqlJks9MbzE8bUIwN_Lphr7y47NfR6BGYnr7M 8272
coax/model_updaters/_model_updater_test.py sha256=mtGg2ZLi9g_ONSJI6bqd6Y8dms7JjDOBh2HnFDl2uj4 3182
coax/policy_objectives/__init__.py sha256=Wy0FOAS4Id2IJI9tj81VTlROoc7wa_UpQj3QOcaFWV8 873
coax/policy_objectives/_base.py sha256=dsgrMCZ3MsbFa2vkP17vf35yF9EZTnM58RFvaX6UflI 7993
coax/policy_objectives/_deterministic_pg.py sha256=u3IeupH5pmgLDIxy3X9IxNqGNBxHQoNto8oB2L6z3tY 5813
coax/policy_objectives/_deterministic_pg_test.py sha256=ZSD9--hRPZNKloXmTof4MPS9bAAmurpWLa5CP-nHdww 1802
coax/policy_objectives/_ppo_clip.py sha256=5GEOKXIIQwJgG1Ja_8mgrRbz5JJMkKVC4vt2YcH8cog 2900
coax/policy_objectives/_ppo_clip_test.py sha256=jjxAz-r-_p1MSKoaFdHEPCKCFKN31AmBx0Vj1JVrWys 1603
coax/policy_objectives/_soft_pg.py sha256=AebxuA4oddQeeSZJS05cTLMngBuorkEbSqAENJIy1ys 4604
coax/policy_objectives/_soft_pg_test.py sha256=fvCliaMpldw-jdcOIqdwF6ZL-nQlKZeRGcol9wFCIs4 4226
coax/policy_objectives/_vanilla_pg.py sha256=X1bQQYnb_8CmdI9KxEU-kB_Pvwzn-4YstrUclUipCKI 1774
coax/policy_objectives/_vanilla_pg_test.py sha256=23QlvuvpEq1FsBT5KhaV7LTAAvO8yEhHu8TJ8DioCFY 4663
coax/proba_dists/__init__.py sha256=ScE-f32CqPm_UHa7jrwgA6uz7SBjYULxC59gfSnjl7I 1198
coax/proba_dists/_base.py sha256=v6TP_8-32pYh41gkAJvZbTZoypmCdSc6YyJo2_d6SH0 8223
coax/proba_dists/_categorical.py sha256=xGzYeXrQ27nVrRwfGdUxwzFJHOrOU2iW8d6E7UWofaY 10618
coax/proba_dists/_composite.py sha256=iO82ZbxPbvZFXoHRGgfVs-qK-LHwRGYzihWVlIU8GaE 11802
coax/proba_dists/_composite_test.py sha256=ufQm-9P2UG_Xyrp86XhRNXACGPZtR1qUuAI3bchfN-4 12738
coax/proba_dists/_discretized_interval.py sha256=LcCZtgQVxWaIKRF7FoNf1wJQqMG-i-g5boBIRwtJlQg 6168
coax/proba_dists/_empirical_quantile.py sha256=d04B6ghRR9Z_Ywx6Xn_EmOtHsqWby0LvkDS0CNhKwiw 2964
coax/proba_dists/_normal.py sha256=Nuxkfe0SpUZa9Hm6Ib_6lrZ7h-WOEvSMDxAOs96u1fY 13182
coax/proba_dists/_normal_test.py sha256=4jMIZaLZCdlG1WRkjtpFdwFpvWOQJP9wYuNPnptOcaY 1443
coax/proba_dists/_squashed_normal.py sha256=9NuW1ilVyn9Xl0YXBRRtzK1wcBGa9DgFdIwRI0NfynY 7209
coax/regularizers/__init__.py sha256=CdDxnD1rGm00PsiJwx--uKh283q-L2gC-O2HZrfsW8M 789
coax/regularizers/_base.py sha256=AEu4S6OdGhP2CujMbIybjWP8e-KQ-JMtNKnBgwBwuuA 2979
coax/regularizers/_entropy.py sha256=r4enBG-hC-z2Rc7GFlcoPfR-DcVhtjMSwYmuvb3w8YA 2326
coax/regularizers/_kl_div.py sha256=DeVZEOPy0N9PpwJnEozdNdIZY3X0MrzJNNbCyLr1km8 3192
coax/regularizers/_nstep_entropy.py sha256=uz1Nu5qpYVAPTSz-l2eYWBI6oyK7leTx5DTXsokXd-A 5091
coax/reward_tracing/__init__.py sha256=kuaoBYsox8qhU68Fe9nkJGNcP6wINkH3qGj1DrCzcA4 1509
coax/reward_tracing/_base.py sha256=9iy6nG_saTck91jM_fKwT3Q19tdfn8uqyaYD04YfaS8 1848
coax/reward_tracing/_montecarlo.py sha256=Qm2qbf2SkfeZN7GBV-r9mjZXTMo6as1eKumU7xUTxXU 1854
coax/reward_tracing/_montecarlo_test.py sha256=S2q_RX3Jad4gpcFoafyXY7o2XOftNKceb8nfUPLeYzE 3645
coax/reward_tracing/_nstep.py sha256=DMVLR01WbfZ5HKzISGOzCI-EADVdTD2FLlzi4rtWdyw 4059
coax/reward_tracing/_nstep_test.py sha256=Ty_1w8jzYhhQ7Cng1ruYEZPLJ9_AuF8XZxFFX3a38kw 11381
coax/reward_tracing/_transition.py sha256=BuHUO_iuHAGLWeSPtbC8XMXPyIVpfMAWvgfJePqRjvQ 7570
coax/td_learning/__init__.py sha256=7XMahTrCxvEzalu8LrUKT0CXZ603SF3cg_tCKngg_Po 1702
coax/td_learning/_base.py sha256=KY-ny3p18Fvm00vQq-Xnfr0sS4HOshGV3r4xfoYsRjQ 22895
coax/td_learning/_clippeddoubleqlearning.py sha256=kN2FoMQ1vzBkHQ1w2v8i14_cgm6A7BgoYhmZkpjOMfA 17026
coax/td_learning/_clippeddoubleqlearning_test.py sha256=_J1YF9aoV5cvrlsDRa7M0X6-MMQfF1OJYGc5J5Xod-0 8951
coax/td_learning/_doubleqlearning.py sha256=puPQIJ2OZsT_cWA-A6VINJ8Ff_Bg91SMDC9CkevCIfU 5168
coax/td_learning/_doubleqlearning_test.py sha256=ONb0tCCFvPveH2ATjic6ozDagil_8FQQrhCTekvzK80 3070
coax/td_learning/_expectedsarsa.py sha256=46eHEvuQai15e1VdO1IhHJ5q1vRPZABtPAwnYPHnVYU 4207
coax/td_learning/_expectedsarsa_test.py sha256=hhIrHXdY0yPZbvEBzPZZA3l5k72yJn2_dYUqYJ7RYTQ 4302
coax/td_learning/_qlearning.py sha256=7k93v6HC6W-4QORy1XfnVldVr-FcA7QvKkjsnOyyt0g 5556
coax/td_learning/_qlearning_test.py sha256=qP0IndlhqWzBpadTklFeS2cWpAB-ZbWfyVpm_SCjJUA 3016
coax/td_learning/_sarsa.py sha256=ZtB2h0WHPoCgSc-nLl7aLEiYvoSu4hDgqEI4nw0f0ZI 2826
coax/td_learning/_sarsa_test.py sha256=x2CsykdG9qIh2eSK3IDQRwQHGDovL2aobZ8R4hsTMGs 5906
coax/td_learning/_simple_td.py sha256=jWdvgxW0xgMxb9mxjGzsvwTf6E8zvLvkEJyu74-2Ofw 2704
coax/td_learning/_simple_td_test.py sha256=xGcpMp7ESoxvbb2AsSCG6ho6tRGiXBfW7owpc9X8FCY 6622
coax/td_learning/_softclippeddoubleqlearning.py sha256=t682UHHfZTgcmWmHFYk_jCW3imgFKzoeoUXiMAaW0ps 6368
coax/td_learning/_softqlearning.py sha256=ECtGedg0X17JXrAP_IWk1baDjJojBRF41njUhGqnKEg 3698
coax/td_learning/_softqlearning_test.py sha256=cVb0LeqY2EfhXNkv_SvozaizUOMu6Hp6DjOAtVolU9k 3958
coax/utils/__init__.py sha256=J4y9n7ZoSgkOzj1Mnpb-FH3PBp0tO88oti_qUz2-OLY 6051
coax/utils/_action_noise.py sha256=Cv_zQfqqiTz0-nuky6-opt36Lr7l676SbLMulgRXimw 3119
coax/utils/_action_noise_test.py sha256=3gsONh9TbR8O2EpGMXfqETUPeKxFucQPUi_APlZooOg 498
coax/utils/_array.py sha256=sHs6AIFhG0pi-xyWTXGe3QV3ieLqIW9TOQuhHzVfpwg 34240
coax/utils/_array_test.py sha256=FLoWjymTYY1MB0j50QUv4HMmTxIo6iDVNzjc4cGnvYo 5291
coax/utils/_array_test_unvectorize.py sha256=yL6LBgMesPh_O8Isb_ixKKrrXHn0VLWrwr8sxbPRvhA 2542
coax/utils/_dmc_gym.py sha256=Jn00krdcx-vAykEEepqliiksBDyZTS-663eTuHbUYZs 3936
coax/utils/_jit.py sha256=Kpp3mc108r37rPOx21C7k5I8PPq4lygWF-Y_HkR15Qo 2163
coax/utils/_misc.py sha256=1Gu7LcDUifRv-IjWBG3emP3WXSv_q8ngbxD8EaMuZUY 18988
coax/utils/_misc_test.py sha256=vy_Usw9zaGs5x-pylXHO4Lfq8MEZK3U165qOyoO0zhM 1377
coax/utils/_quantile_funcs.py sha256=XBvB00JpB7NBi8FdWsOsZXyFsP6WJpg4p9O63sWOUlQ 2764
coax/utils/_rolling.py sha256=NWkzf0csytAqaM3C2MS6sZkgZmg6iboxoPwH0Q-LLDQ 992
coax/utils/_segment_tree.py sha256=-YxdrRurr86yikHokD49vKnoYNYislMAKoMYaErCy40 15361
coax/utils/_segment_tree_test.py sha256=UOYYg9_Ra22A3_O1C5ArnTDIf6ct5XsGuJ5qKp19gQ8 4278
coax/value_losses/__init__.py sha256=uaw4CMWbgBxye0jcswrK2J2tMQ6FQuPkAHSOjrhJk7g 825
coax/value_losses/_losses.py sha256=N76e3nSE6UIA0GioByVhBq3wXCOfznoimN4Uidn1xRo 5578
coax/value_transforms/__init__.py sha256=HXqe-X5iWMNfXF0sno8eI7FZziIoZGGoo1TrMJfzqR0 659
coax/value_transforms/_base.py sha256=2U-zLBVhdAkMuAEYQXT-H1uAN6XyZQOoY33ol64xpTw 1263
coax/value_transforms/_log_transform.py sha256=SJepZv1Bvyw9VoqYYSGuL12e9iE_YfZR4SCHl8yBnZQ 1385
coax/value_transforms/_log_transform_test.py sha256=TWZvQ39xsW4qR1teQnjSjpNpMZ3jn0o1WJn52fXzi88 429
coax/wrappers/__init__.py sha256=Rr0IhoVhyDlcBQmA651Vdpzb5ZCw87phfFKq6ePEdvg 1366
coax/wrappers/_box_spaces.py sha256=DhAK5d4E1eWa9qBwrR5GOoVg8i_9V2yNgzpGcR6nIwM 4466
coax/wrappers/_box_spaces_test.py sha256=vRo-IpNGzMbGHaHnkbXrxBreHA1QvKkW8246p3JF7pM 1259
coax/wrappers/_frame_stacking.py sha256=c_uryLea7IHClBMxvf76fjgt3zAhxvA8C_g6qlQsglw 2186
coax/wrappers/_meta_policy.py sha256=yLe3OONyOpjXaNGpDH1LC_q9vXRYgVrHsQSpJQu0_jU 1824
coax/wrappers/_train_monitor.py sha256=GcfEpw6gAFl-Us5Ai8DxG8ixkQX7N4nARk_EoNLQyZA 12563
coax-0.1.13.dist-info/LICENSE sha256=GUO5sWd3jPTL6of1wYkfmt8HjIDaaj05U8yUVRBjwQc 1189
coax-0.1.13.dist-info/METADATA sha256=PVCPYLoIfcPnoj47CYAkkcgYTmpxWGUTwxN3dnjbNUM 4333
coax-0.1.13.dist-info/WHEEL sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo 92
coax-0.1.13.dist-info/top_level.txt sha256=1J5SaBKdLC1Xct9i44KsqpjdWz7xZj4VeioEaM4o5zM 5
coax-0.1.13.dist-info/zip-safe sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
coax-0.1.13.dist-info/RECORD

top_level.txt

coax

zip-safe