search

luchris429 / purejaxrl

Really Fast End-to-End Jax RL Implementations
Apache License 2.0
738 stars 62 forks source link

[Docs] Mention request for JAX-CORL #24

Closed nissymori closed 5 months ago

nissymori commented 6 months ago

Dear purejaxrl team,

First of all, I would like to express my appreciation for developing the fascinating codebase, purejaxrl The high-quality single-file implementation in JAX is truly impressive, and I have greatly enjoyed using it, especially PPO(really fast).

As a researcher who predominantly uses JAX due to its speed and efficiency, I recognized the potential value of a offline counterpart of purejaxrl. To address this gap, I recently launched JAX-CORL, which provides single-file implementations of Offline RL algorithms specifically in JAX.

I believe that purejaxrl and JAX-CORL can complement each other as online and offline versions of clean RL in JAX. If you find it suitable, I would be happy if you could mention JAX-CORL in the related work section of your README.

Thank you!

luchris429 commented 5 months ago

That sounds great! Could you open a PR for it?

luchris429 commented 5 months ago

Actually I just added it. Great work!

nissymori commented 5 months ago

@luchris429 Thank you for your mentioning!!!