ppo_jax icon indicating copy to clipboard operation
ppo_jax copied to clipboard

Published 20 hours ago • verified publisher icon bmazoure

Training speed and computation cost

Open kaixin96 opened this issue 2 years ago • 0 comments

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 and GPU memory consumption.

Look forward to your reply!

kaixin96 avatar Aug 24 '22 18:08 kaixin96