search

luchris429 / purejaxrl

Really Fast End-to-End Jax RL Implementations
Apache License 2.0
615 stars 53 forks source link

Recurrent network support #3

Closed LianhaoYin closed 1 year ago

LianhaoYin commented 1 year ago

It will be great to support the recurrent network in the PPO

luchris429 commented 1 year ago

That's a good idea! I have some code for that, though I need to clean it up and package it for this repository!

luchris429 commented 1 year ago

This should be completed now!

https://github.com/luchris429/purejaxrl/blob/main/purejaxrl/ppo_rnn.py