Wheelodex » Projects » dopamax » dopamax-0.2.1-py3-none-any.whl

dopamax

View on PyPI — Reverse Dependencies (0)

0.2.1 dopamax-0.2.1-py3-none-any.whl

Wheel Details

Project: dopamax
Version: 0.2.1
Filename: dopamax-0.2.1-py3-none-any.whl
Download: [link]
Size: 52336
MD5: 0ab4eb0fb8b7dfb3c9ef09d595fc1057
SHA256: 2124e8f41c9184e70d583f2dfa771dbb76e7c02f529738f324db407ef32731f9
Uploaded: 2025-01-12 21:25:35 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt

METADATA

Metadata-Version: 2.2
Name: dopamax
Version: 0.2.1
Summary: Reinforcement learning in pure JAX.
Author-Email: Ryan Strauss <ryanrstrauss[at]icloud.com>
Project-Url: Repository, https://github.com/rystrauss/dopamax.git
License: MIT License Copyright (c) 2024 Ryan Strauss 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: Development Status :: 3 - Alpha
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: jax (==0.4.38)
Requires-Dist: jaxlib (==0.4.38)
Requires-Dist: chex (>=0.1.87)
Requires-Dist: brax (>=0.11.0)
Requires-Dist: click (>=8.1.7)
Requires-Dist: distrax (>=0.1.5)
Requires-Dist: dm-env (>=1.6)
Requires-Dist: dm-haiku (>=0.0.13)
Requires-Dist: pgx (>=2.4.2)
Requires-Dist: einops (>=0.8.0)
Requires-Dist: ffmpeg (>=1.4)
Requires-Dist: imageio (>=2.25.1)
Requires-Dist: mctx (>=0.0.5)
Requires-Dist: moviepy (>=1.0.3)
Requires-Dist: ml-collections (>=0.1.1)
Requires-Dist: optax (>=0.2.4)
Requires-Dist: numpy (>=1.22.4)
Requires-Dist: rlax (>=0.1.6)
Requires-Dist: tqdm (>=4.64.1)
Requires-Dist: wandb (>=0.18.7)
Requires-Dist: flashbax (>=0.1.2)
Requires-Dist: gymnax (>=0.0.8)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4794 characters]

WHEEL

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

RECORD

Path Digest Size
dopamax/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dopamax/callbacks.py sha256=WBYG6UO8jaTecPUFaKU5gZaJMknVxM8vulgQHa3eP4w 1328
dopamax/distributions.py sha256=5uVUJGfTcFB1HuWzdtQvVILaXVtykxuYPyxKoJZPf4M 254
dopamax/networks.py sha256=7HGyVIobLMUhp3_KbIUBjuVZBeE3QbBq_qlcGEfFMXY 12564
dopamax/prioritized_item_buffer.py sha256=P4TLrr8SCNKUqCzSBv5u7-Yxj24rWNSkHo2DoHPa4nA 5939
dopamax/rollouts.py sha256=ZAH1G3edZWauXIXzsQqeOr1CG3-gZckIS6zOEjflXso 7245
dopamax/spaces.py sha256=dcZMBUzv_NjHU8gAYxmJS5jB9wOu0Qv76dSxTgvAXQg 6283
dopamax/typing.py sha256=O3upwXR1KiyzSDDeolVlUw_lrzxBCycau7p7BxIwJLA 259
dopamax/utils.py sha256=YYLHjR3Qq1cFIDCTD3-z9x0OyTy9-MxesBXDPEUTjME 1273
dopamax/_scripts/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dopamax/_scripts/cli.py sha256=tSsFgbfSEAVTHAUTl7eVUXT9NvrHMhpShDPKjvaDZiY 5784
dopamax/agents/__init__.py sha256=cYmJXYfu8G4gnV49JDi4bF6Cq0dYRvnG3G9_7jEXRN0 159
dopamax/agents/base.py sha256=seOLcJkexuXvELNSnyxSb7BTRl5aaj_PJmGdij05AVw 4510
dopamax/agents/utils.py sha256=1fwQHad6i3jJ2PiAjW0I_BbGE_AUGnXP84TVZALcM20 412
dopamax/agents/anakin/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dopamax/agents/anakin/alphazero.py sha256=XNUmf_a94MJaRpykZxp5Kub6M03giu_rbgdAx-K76ug 13159
dopamax/agents/anakin/base.py sha256=2s77AHwHB5ITvU9p1deN-9TZpWAAoEYR1L93lRkN_rQ 15098
dopamax/agents/anakin/ddpg.py sha256=q6UfZUC66ALuKhFOVgXF5oXMJiTQDkKjMq5sa36aDNk 16526
dopamax/agents/anakin/dqn.py sha256=iN-x4Berk5uICVY5tMcKoxOZedoQO7RT4jU2hCzs_P4 12537
dopamax/agents/anakin/ppo.py sha256=wSHkapJtp3EcbZ-1UNPNuBYnXoGqiUEwlthrQztZZho 9873
dopamax/agents/anakin/sac.py sha256=edNEHlN-exLlWH6AAh1PYyBeb1wINNfRMbMQxltiC7s 17551
dopamax/agents/anakin/utils.py sha256=aCGfEb0PNW7zkMYT8KUAfqsawc0XFtCH6gvRp7G2TQ8 1642
dopamax/environments/__init__.py sha256=FC_u9qx3Ku505dZchXRzaYZ4oJRGYE2rbSmXH-SbQuQ 150
dopamax/environments/environment.py sha256=xL2OFG3AqjVcszUUgdzDC4EVzFMDvf7vseGWCoCRk5Y 5987
dopamax/environments/gymnax.py sha256=utvI_cpgCxQYIcMkI3dZL6_RC2TU5xSXx1XlxnPmOp8 2559
dopamax/environments/utils.py sha256=G4Z89EXFxg_vE21y-uODhnVW-9H_oFRklhEBgDPjTNQ 975
dopamax/environments/visualization.py sha256=rRb6A7mg7iMrUipJVPP1OXnrzWsko6KIkwxmlnGK2t8 898
dopamax/environments/brax/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dopamax/environments/brax/ant.py sha256=FELlOM06aw5oUmvnzGyJUYGiIU6NUHztTLNc4M52xpc 821
dopamax/environments/brax/base.py sha256=6C6paxNP9-mRqFBLrk1OZqz8bzb69gkwIlZU5WTbQLc 1920
dopamax/environments/brax/half_cheetah.py sha256=N1SRidWWhOL4yB6HuSCWM-iX3b71V1l7CBzabdas9V0 853
dopamax/environments/brax/inverted_pendulum.py sha256=ik7BjKqEbgHgoznQCNonoQDDw1J8hBsrNCJ41Y70QLU 880
dopamax/environments/pgx/__init__.py sha256=Hceu_xjo5kiGTrpqRAMnsjXxvtNDf8HnCqo0P_uowto 104
dopamax/environments/pgx/base.py sha256=SwqOYp0oxcb1r742jpwLxY3IPXHTWQkJMFBzhv4Gpoo 3278
dopamax/environments/pgx/chess.py sha256=U1R0NtjaoyeuIVCXKzMzh3t5vT1g2ENtu3C0atnrjWg 509
dopamax/environments/pgx/connect_four.py sha256=JvaEQTctV-uW2biu4NzzFmYoTwvpaEX1Oeg5RYwagQw 512
dopamax/environments/pgx/go.py sha256=v7b5zncMy4DNKYtmCBx9217cVJa-CffZEtLhuxzRw1s 879
dopamax/environments/pgx/tic_tac_toe.py sha256=w7P0L6uaqkL1nXWb74MIsNLH8Loxr65XgVFoTdDsr1U 501
dopamax-0.2.1.dist-info/LICENSE sha256=E3IsiFRd8rRAUWRQBRJzb5pIEavZCfvZ9FJ04jzNEOU 1069
dopamax-0.2.1.dist-info/METADATA sha256=0d2wIgL8byLfay3Ne3NpWc5vgzFJ3UxWSVwX0KCVcmM 7300
dopamax-0.2.1.dist-info/WHEEL sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8 91
dopamax-0.2.1.dist-info/entry_points.txt sha256=BklSmSmk03jSt-VPtJZPgc9c00CmOG2GFgKhS2rAuvI 53
dopamax-0.2.1.dist-info/top_level.txt sha256=C1I6YcpdWtZLDIE8O-TFMUMHRAB1DaHYBS3N7PwKHr8 8
dopamax-0.2.1.dist-info/RECORD — —

top_level.txt

dopamax

entry_points.txt

[console_scripts]
dopamax = dopamax._scripts.cli:cli