Wheelodex » Projects » dopamax

dopamax

View on PyPI — Reverse Dependencies (0)

0.2.0 dopamax-0.2.0-py3-none-any.whl

Wheel Details

Project: dopamax
Version: 0.2.0
Filename: dopamax-0.2.0-py3-none-any.whl
Download: [link]
Size: 49586
MD5: f08f04a97382da6ff2fa592d483c4d7d
SHA256: 9e0787b56036cb968ec541182e7292cbb9baeb2c265198ed6824fdcdd265e748
Uploaded: 2023-12-29 21:20:52 +0000

dist-info

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

METADATA

Metadata-Version: 2.1
Name: dopamax
Version: 0.2.0
Summary: Reinforcement learning in pure JAX.
Author: Ryan Strauss
Author-Email: ryanrstrauss[at]icloud.com
Home-Page: https://github.com/rystrauss/dopamax
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Platform: any
Requires-Python: >=3.9
Requires-Dist: jax (>=0.4.23)
Requires-Dist: jaxlib (>=0.4.23)
Requires-Dist: chex (>=0.1.6)
Requires-Dist: brax (>=0.9.0)
Requires-Dist: click (>=8.1.3)
Requires-Dist: distrax (>=0.1.3)
Requires-Dist: dm-env (>=1.6)
Requires-Dist: dm-haiku (>=0.0.9)
Requires-Dist: pgx (>=2.0.1)
Requires-Dist: einops (>=0.6.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.1.7)
Requires-Dist: pygame (>=2.1.3)
Requires-Dist: numpy (>=1.22.4)
Requires-Dist: rlax (>=0.1.5)
Requires-Dist: tqdm (>=4.64.1)
Requires-Dist: wandb (>=0.13.10)
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4373 characters]

WHEEL

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

RECORD

Path Digest Size
dopamax/__init__.py sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0 22
dopamax/callbacks.py sha256=klTv69fvjwHtsLDm9eosXSGKPZyvGE15zf0vYR3lBog 1328
dopamax/networks.py sha256=tQtd4IL6dn4MgqimunHp0CIqRHpZUpkx930LxcgMXms 9137
dopamax/rollouts.py sha256=ZP6SB0245kECJKdLF-wPIkcSLRHop9EwfMisBTCQ_kQ 7496
dopamax/spaces.py sha256=Ccfx6M4WGSdlh-tOmx41zFj_p3pjaeLdRwRCJQe0Org 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=CzKRi81jL7kHYLyPdl6D0zBTNremPiFXGY7zEISAeOU 5631
dopamax/agents/__init__.py sha256=BvHQM0tBaRxRhHVishZlLcHxT1SVeWY8iYHPvls1pjY 96
dopamax/agents/base.py sha256=MncxfPkQiWdPwBmF0_9HT_gXVsDQc0qaes3NuSrsXL8 4516
dopamax/agents/utils.py sha256=r8wxe0r9LqxGBmophmAj6Zk7W1qpf_4pPGscFZdbIQs 466
dopamax/agents/anakin/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dopamax/agents/anakin/alphazero.py sha256=rCOjTqNVUsLy9FgjEbJgZAj2JnPsnPpcyCeYfQfTAto 13055
dopamax/agents/anakin/base.py sha256=B0DNc8y2RZ-lKX5-em_CUTmhbW1ZVl579dl6iXH6d1Y 14851
dopamax/agents/anakin/dqn.py sha256=v0tYL19Frj2KznVtEwFXJy1UuQjBtAbdJiej2fG2dO0 10126
dopamax/agents/anakin/ppo.py sha256=LGobHig5GgDhtbyHIQe0voSi3pMXHm48sHp1HzyX6D8 9721
dopamax/agents/anakin/utils.py sha256=aCGfEb0PNW7zkMYT8KUAfqsawc0XFtCH6gvRp7G2TQ8 1642
dopamax/environments/__init__.py sha256=klN6dmOZrn8SYFzWNH9NHBUZJna7nKtwZ97PayTOK2g 334
dopamax/environments/cartpole.py sha256=6NKOCY2jg7xaCOdVcdH8zls6Db0LBg_rERm2sanT8-g 6855
dopamax/environments/environment.py sha256=KLuxPX2i7T0vmP4DLptxywBrZEqNCzlf8BmXMtJM3DE 6802
dopamax/environments/mountain_car.py sha256=UF0MeKIVCkw3ACRzFuy-8ChvQ-ckIgApAjiUeJ2fvx4 6440
dopamax/environments/mountain_car_continuous.py sha256=G5TZPu-QSXhdjaRH3IphxqfFaEXeHNXYNeh-zAMSDp4 6698
dopamax/environments/utils.py sha256=j4cPzC4Lkvvb4IUq8tE0yhorgOBkke7GDvSD9FEOJCo 569
dopamax/environments/brax/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
dopamax/environments/brax/ant.py sha256=OGixYboRmR_6PrpgkFBeWjk88z-1DD_09CTgYotExx8 816
dopamax/environments/brax/base.py sha256=X8wBUuFhCPkQDnmOiF2rzQ_u74S01vNjfFlKBdokvhY 2195
dopamax/environments/brax/half_cheetah.py sha256=mtkFJ08vRUSkos8KCycLm4LmsHZ59szQGYFWORVomdY 848
dopamax/environments/brax/inverted_pendulum.py sha256=6sXyDTs9jvoXlqivkh-RqgiqgBFIKW9sCCbkSI35aRE 875
dopamax/environments/pgx/__init__.py sha256=Hceu_xjo5kiGTrpqRAMnsjXxvtNDf8HnCqo0P_uowto 104
dopamax/environments/pgx/base.py sha256=uUniEIxMx2QQ0_L6iXoektra4YquMCuuyZiBupUuWOw 3277
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
tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/test_rollouts.py sha256=nNrxvUqaDUwI6p4JmVDGX0LbJ_DvxhbI_9QkQJDr6xQ 1123
tests/test_spaces.py sha256=vhx5wm4CxxqtNqun76bnJ1JbsTr77X9S6EyjOHt86cY 2231
tests/environments/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/environments/test_cartpole.py sha256=B4L7MGN4uwjXVtDDf_4H5a5lvTvluq5GusMzxlbWfik 2098
tests/environments/test_mountain_car.py sha256=do783tFAj_8OgggXTShlMCnfbu-cyu4yY_BvzzbSxNY 1763
tests/environments/test_mountain_car_continuous.py sha256=YjoAmiUvdrYMLaLimyjevsEcRwBVTpPsj2NTP_bcJNc 1840
dopamax-0.2.0.dist-info/LICENSE sha256=BPl0hd-T5BRAr6kkT4CtG2TI7VaGzBEaBijXtI9DKuo 1069
dopamax-0.2.0.dist-info/METADATA sha256=Uo7sU7EgVDHE7_ohd_JF6PMRu4QMtJROsWeBmDHBb_s 5623
dopamax-0.2.0.dist-info/WHEEL sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM 92
dopamax-0.2.0.dist-info/entry_points.txt sha256=BklSmSmk03jSt-VPtJZPgc9c00CmOG2GFgKhS2rAuvI 53
dopamax-0.2.0.dist-info/top_level.txt sha256=ZTYQzrqb462HIFNz28UKVC3B6lE4xqzAhFd_X87Yf8k 14
dopamax-0.2.0.dist-info/RECORD — —

top_level.txt

dopamax
tests

entry_points.txt

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