https://github.com/vwxyzjn/envpool-xla-cleanrl

https://github.com/vwxyzjn/envpool-xla-cleanrl

Last synced: about 2 months ago
JSON representation

• Host: GitHub
• URL: https://github.com/vwxyzjn/envpool-xla-cleanrl
• Owner: vwxyzjn
• License: mit
• Created: 2022-07-11T21:54:15.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2022-12-31T16:37:35.000Z (about 2 years ago)
• Last Synced: 2024-11-09T19:04:46.601Z (3 months ago)
• Language: Python
• Size: 26.9 MB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# envpool-xla-cleanrl (Experimental)

This is an experimental support of leveraging envpool's XLA interface with CleanRL's PPO. To get started, run the following command.

```

poetry install

poetry run pip install envpool==0.6.3

poetry run pip install --upgrade "jax[cuda]==0.3.14" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

poetry run python ppo_atari_envpool_xla_jax.py

```