rejax

View on PyPIReverse Dependencies (1)

0.1.1 rejax-0.1.1-py3-none-any.whl

Wheel Details

Project: rejax
Version: 0.1.1
Filename: rejax-0.1.1-py3-none-any.whl
Download: [link]
Size: 36532
MD5: 5c6a4d14a5af3d0021c975bb401de15b
SHA256: 404d2a1dcdd6595924a1d5decb2551e2d5ffb47294c94e7b848f79e3c0e59c39
Uploaded: 2024-11-14 16:37:57 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: rejax
Version: 0.1.1
Summary: Vectorizable RL algorithms in pure JAX
Author: Jarek Liesen
License: Apache-2.0
Requires-Python: >=3.10
Requires-Dist: pyyaml (>=6.0.2)
Requires-Dist: flax (>=0.9.0)
Requires-Dist: optax (>=0.2.3)
Requires-Dist: brax (>=0.11.0)
Requires-Dist: evosax (>=0.1.6)
Requires-Dist: distrax (>=0.1.5)
Requires-Dist: gymnax (>=0.0.8)
Requires-Dist: jax (>=0.4.34)
Requires-Dist: brax (>=0.11.0); extra == "compat"
Requires-Dist: craftax (>=1.4.3); extra == "compat"
Requires-Dist: navix (>=0.7.0); extra == "compat"
Requires-Dist: jumanji (>=1.0.1); extra == "compat"
Provides-Extra: compat
Description-Content-Type: text/markdown
[Description omitted; length: 7423 characters]

WHEEL

Wheel-Version: 1.0
Generator: pdm-backend (2.4.3)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
rejax-0.1.1.dist-info/METADATA sha256=4rAHoaJ185_LHwD2xUhxWbmcL7eQGpKWdC3LhWswENE 8106
rejax-0.1.1.dist-info/WHEEL sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4 90
rejax-0.1.1.dist-info/entry_points.txt sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs 34
rejax-0.1.1.dist-info/licenses/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
rejax/__init__.py sha256=k46Nuwmm_hXKwDsO-huAluMxMAzPca9_gHN6ON60Dy0 387
rejax/algos/__init__.py sha256=902MpClKtTXTqCM_mdI9omhfiUo9Dac0qLYot3V166Q 545
rejax/algos/algorithm.py sha256=i_IKin2OH8KljgBTDmckewDvSLFkEeTvmWek4zXeo6g 3567
rejax/algos/dqn.py sha256=GWNMCBBnzxAfjNcH_qPUf-RReFXAyxvBVEVu4OIoEN4 6927
rejax/algos/iqn.py sha256=ot_Vb_JZt8usOFRsL33ADKoAb8Unoc9_4tm09A5fidg 8687
rejax/algos/mixins.py sha256=WHvW1YE6F3mcfziYvfqraZ3vTacSHVNMB_tNBGU1C7w 7301
rejax/algos/ppo.py sha256=h70VrL2AOSa1fssLtZ31xXXfUhRQ60mh-KXfjKP-rMM 8513
rejax/algos/pqn.py sha256=HXmCR3PVRh7Z0cGww373u6B4x0qyw4UTaW8IFzwBjbw 6039
rejax/algos/sac.py sha256=EW0s0MNN74PyvCNE3ugi6KGEQfKtYjeyFo3JCm8YiFs 10891
rejax/algos/td3.py sha256=n0y0fCUaYaSFbKYhe9-eZWhdbO4e1uZDqi_oGu9Q3pY 11959
rejax/buffers.py sha256=hN8vbDxX_NkO6yzN8aoauuAhlBwUxKGKC1Lg3j7d3tI 3674
rejax/compat/__init__.py sha256=N5tLx2Il534kOMYOmIdOMTGN16ohlzOimgDF770vQm8 579
rejax/compat/brax2gymnax.py sha256=GJt2GSlsSbolZijzHvR-waSnYiOzw71-cZuDb4eM5uM 2081
rejax/compat/jumanji2gymnax.py sha256=gE3M77tEFygFoAHP2SiY1VzFwuec4-BQ0vaHF_AvMpM 5820
rejax/compat/navix2gymnax.py sha256=ePpKTVqh7IxXmwDcbG4xH7U1BLjOpU6QrXJ5_BBcCTQ 1546
rejax/evaluate.py sha256=idoyUTtA1_r9HyqzqtdLqlz7aJhiBuI17TCkHZSlzaA 2691
rejax/networks.py sha256=Ohq1yO9Xp6GkQtMr9rhNessVnnscyX9HXiR8Abu3XcA 10150
rejax/normalize.py sha256=demQlOedAg_kGOrsgmm7teyef3kI7j3oTBVQbqAHNPw 2482
rejax-0.1.1.dist-info/RECORD

entry_points.txt

[empty]