Re:4000x Speedup in Reinforcement Learning with Jax
by nothrowaways on Friday April 07, 2023 @04:33AM
(#35479537)
It is misleading, the speedup is not just because it is Jax. The devil is in the GPU
[ Parent
| on HN ]
Re:4000x Speedup in Reinforcement Learning with Jax
by luchris429 on Saturday April 08, 2023 @11:34AM
(#35494283)
Indeed the devil is in the GPU! Jax and its ecosystem just make it much easier to use the GPU.