pyRDDLGym-jax 0.2__py3-none-any.whl → 0.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (24) hide show
  1. pyRDDLGym_jax/__init__.py +1 -0
  2. pyRDDLGym_jax/core/compiler.py +90 -68
  3. pyRDDLGym_jax/core/logic.py +188 -46
  4. pyRDDLGym_jax/core/planner.py +411 -195
  5. pyRDDLGym_jax/core/simulator.py +2 -1
  6. pyRDDLGym_jax/core/tuning.py +13 -10
  7. pyRDDLGym_jax/examples/configs/HVAC_ippc2023_drp.cfg +3 -3
  8. pyRDDLGym_jax/examples/configs/MarsRover_ippc2023_drp.cfg +1 -0
  9. pyRDDLGym_jax/examples/configs/Pendulum_gym_slp.cfg +1 -1
  10. pyRDDLGym_jax/examples/configs/default_drp.cfg +1 -1
  11. pyRDDLGym_jax/examples/configs/default_slp.cfg +1 -1
  12. pyRDDLGym_jax/examples/run_gym.py +2 -5
  13. pyRDDLGym_jax/examples/run_plan.py +6 -8
  14. pyRDDLGym_jax/examples/run_scipy.py +61 -0
  15. pyRDDLGym_jax/examples/run_tune.py +5 -6
  16. pyRDDLGym_jax-0.4.dist-info/METADATA +276 -0
  17. {pyRDDLGym_jax-0.2.dist-info → pyRDDLGym_jax-0.4.dist-info}/RECORD +20 -22
  18. {pyRDDLGym_jax-0.2.dist-info → pyRDDLGym_jax-0.4.dist-info}/WHEEL +1 -1
  19. pyRDDLGym_jax/examples/configs/Pong_slp.cfg +0 -18
  20. pyRDDLGym_jax/examples/configs/SupplyChain_slp.cfg +0 -18
  21. pyRDDLGym_jax/examples/configs/Traffic_slp.cfg +0 -20
  22. pyRDDLGym_jax-0.2.dist-info/METADATA +0 -26
  23. {pyRDDLGym_jax-0.2.dist-info → pyRDDLGym_jax-0.4.dist-info}/LICENSE +0 -0
  24. {pyRDDLGym_jax-0.2.dist-info → pyRDDLGym_jax-0.4.dist-info}/top_level.txt +0 -0
pyRDDLGym_jax/examples/configs/SupplyChain_slp.cfg DELETED
@@ -1,18 +0,0 @@
1
- [Model]
2
- logic='FuzzyLogic'
3
- logic_kwargs={'weight': 10.0}
4
- tnorm='ProductTNorm'
5
- tnorm_kwargs={}
6
-
7
- [Optimizer]
8
- method='JaxStraightLinePlan'
9
- method_kwargs={}
10
- optimizer='rmsprop'
11
- optimizer_kwargs={'learning_rate': 0.005}
12
- batch_size_train=8
13
- batch_size_test=8
14
-
15
- [Training]
16
- key=42
17
- epochs=10000
18
- train_seconds=90
pyRDDLGym_jax/examples/configs/Traffic_slp.cfg DELETED
@@ -1,20 +0,0 @@
1
- [Model]
2
- logic='FuzzyLogic'
3
- logic_kwargs={'weight': 1000}
4
- tnorm='ProductTNorm'
5
- tnorm_kwargs={}
6
-
7
- [Optimizer]
8
- method='JaxStraightLinePlan'
9
- method_kwargs={}
10
- optimizer='rmsprop'
11
- optimizer_kwargs={'learning_rate': 0.001}
12
- batch_size_train=16
13
- batch_size_test=16
14
- clip_grad=1.0
15
-
16
- [Training]
17
- key=42
18
- epochs=200
19
- train_seconds=30
20
- policy_hyperparams={'advance': 10.0}
pyRDDLGym_jax-0.2.dist-info/METADATA DELETED
@@ -1,26 +0,0 @@
1
- Metadata-Version: 2.1
2
- Name: pyRDDLGym-jax
3
- Version: 0.2
4
- Summary: pyRDDLGym-jax: JAX compilation of RDDL description files, and a differentiable planner in JAX.
5
- Home-page: https://github.com/pyrddlgym-project/pyRDDLGym-jax
6
- Author: Michael Gimelfarb, Ayal Taitler, Scott Sanner
7
- Author-email: mike.gimelfarb@mail.utoronto.ca, ataitler@gmail.com, ssanner@mie.utoronto.ca
8
- License: MIT License
9
- Classifier: Development Status :: 3 - Alpha
10
- Classifier: Intended Audience :: Science/Research
11
- Classifier: License :: OSI Approved :: MIT License
12
- Classifier: Natural Language :: English
13
- Classifier: Operating System :: OS Independent
14
- Classifier: Programming Language :: Python :: 3
15
- Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
16
- Requires-Python: >=3.8
17
- License-File: LICENSE
18
- Requires-Dist: pyRDDLGym >=2.0
19
- Requires-Dist: tqdm >=4.66
20
- Requires-Dist: bayesian-optimization >=1.4.3
21
- Requires-Dist: jax >=0.4.12
22
- Requires-Dist: optax >=0.1.9
23
- Requires-Dist: dm-haiku >=0.0.10
24
- Requires-Dist: tensorflow >=2.13.0
25
- Requires-Dist: tensorflow-probability >=0.21.0
26
-