ppo_jax icon indicating copy to clipboard operation
ppo_jax copied to clipboard

Published 20 hours ago • verified publisher icon bmazoure
→

Metadata

Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.

Results 2 ppo_jax issues
Sort by recently updated
recently updated
newest added

Training speed and computation cost

Hi @bmazoure , thank you for this great repo! Just curious about how JAX compares to TensorFlow 1.x (the official OpenAI's train-procgen repo) and PyTorch, in terms of training speed...

kaixin96 avatar kaixin96

PPO's performance

1 comment

Hi @bmazoure, Your PPO +JAX implementation caught my eyes and this is a really cool repo! Based on your [benchmark](https://wandb.ai/bmazoure/ppo_procgen_jax/reports/PPO-Procgen-JAX-version---VmlldzoxMDM4MjAx) with W&B, I compared the performance of your implementation with...

vwxyzjn avatar vwxyzjn

← Metadata

47
Stars
1
Forks
Watchers

Owner

verified publisher icon bmazoure

Metadata

Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.

Back