jaxagents

View on PyPIReverse Dependencies (0)

0.1.7 jaxagents-0.1.7-py3-none-any.whl

Wheel Details

Project: jaxagents
Version: 0.1.7
Filename: jaxagents-0.1.7-py3-none-any.whl
Download: [link]
Size: 39136
MD5: ce5d4e4bdd24442c6275de54d818f7c3
SHA256: da42da9a5d52adc8d3603a3d582c87f271fab59eb5733f474dff1a07532cf883
Uploaded: 2024-08-25 13:53:39 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jaxagents
Version: 0.1.7
Summary: Implementation of Reinforcement Learning agents in JAX
Author: Antonis Mavritsakis
Author-Email: amavrits[at]gmail.com
Home-Page: https://github.com/amavrits/jax-agents
License: MIT
Requires-Python: >=3.9.0
Requires-Dist: jaxlib (==0.4.28)
Requires-Dist: jax (==0.4.28)
Requires-Dist: chex (==0.1.86)
Requires-Dist: distrax (==0.1.5)
Requires-Dist: flax (==0.8.4)
Requires-Dist: flashbax (==0.1.2)
Requires-Dist: jaxtyping (==0.2.29)
Requires-Dist: gymnax (==0.0.6)
Requires-Dist: jax-tqdm (==0.2.1)
Requires-Dist: optax (==0.2.2)
Requires-Dist: orbax-checkpoint (==0.5.14)
Requires-Dist: numpy (==1.26.4)
Requires-Dist: scipy (==1.13.1)
Requires-Dist: matplotlib (==3.9.0)
Requires-Dist: pytest (==8.2.1)
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE.txt
[Description omitted; length: 794 characters]

WHEEL

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

RECORD

Path Digest Size
jaxagents/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxagents/dqn.py sha256=iK9Zl5FJF14VNCgo0jiEmlacRKwhzVZUGEvMMVSD5Uo 63362
jaxagents/ppo.py sha256=5aSy3N_Qk7NrIZDE0oaoTKImn4cFjjXifxyiTJfNaPs 57112
jaxagents/vpg.py sha256=76qSBzussZziJXiapWO-KMxjMMQQxgrhD7uJS2pLMr4 40335
jaxagents/agent_utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jaxagents/agent_utils/dqn_utils.py sha256=5CrdbuBKiF0LPm7UkTbnoq5JxGOecx1EFS0SDJiP6ho 7013
jaxagents/agent_utils/ppo_utils.py sha256=HGxRc8SHYq5ktuBzL47TP2ZFrEnkqIG6fnjUY__xT7Q 6795
jaxagents/agent_utils/vpg_utils.py sha256=STd_LImLUURAAtPlk_QCGGH6zJfBeRv1bvdnSwaAypI 5089
jaxagents-0.1.7.dist-info/LICENSE.txt sha256=4HBVyDiGNK87tJC4StPe9skRQ_MU1hfhbRFywpIkaf0 1083
jaxagents-0.1.7.dist-info/METADATA sha256=FG03gki43vesoqE6v5Aw8S72NZYhk48rZpi3-fNhQls 1666
jaxagents-0.1.7.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
jaxagents-0.1.7.dist-info/top_level.txt sha256=tXeaz3hkqqrG64BJK6NGORCvcvAnYuB1VOrm8MFMRes 10
jaxagents-0.1.7.dist-info/RECORD

top_level.txt

jaxagents