Adds wrappers for jax environments converting to numpy and pytorch (#168)

Co-authored-by: Justin Deutsch <djustin8@vt.edu>
Co-authored-by: Gianluca De Cola <42657588+gianlucadecola@users.noreply.github.com>
This commit is contained in:
Mark Towers
2022-12-03 19:46:01 +00:00
committed by GitHub
parent e2caec7c06
commit 9157a97c80
7 changed files with 503 additions and 8 deletions

View File

@@ -175,21 +175,21 @@ Gymnasium already contains a large collection of wrappers, but we believe that t
* - :class:`wrappers.EnvCompatibility`
- Moved to `shimmy <https://github.com/Farama-Foundation/Shimmy/blob/main/shimmy/openai_gym_compatibility.py>`_
- Not Implemented
* - :class:`RecordEpisodeStatistics`
* - :class:`wrappers.RecordEpisodeStatistics`
- RecordEpisodeStatistics
- VectorRecordEpisodeStatistics
* - :class:`RenderCollection`
* - :class:`wrappers.RenderCollection`
- RenderCollection
- VectorRenderCollection
* - :class:`HumanRendering`
* - :class:`wrappers.HumanRendering`
- HumanRendering
- Not Implemented
* - Not Implemented
- JaxToNumpy
- VectorJaxToNumpy
- :class:`experimental.wrappers.JaxToNumpy`
- VectorJaxToNumpy (*)
* - Not Implemented
- JaxToTorch
- VectorJaxToTorch
- :class:`experimental.wrappers.JaxToTorch`
- VectorJaxToTorch (*)
```
### Vector Only Wrappers