Re:4000x Speedup in Reinforcement Learning with Jax
by m00x on Thursday April 06, 2023 @11:29PM
(#35477698)
I don't get that sense at all.
You could do the same with tensorflow and pytorch, but in my experience, with more difficulty since they're more opinionated about how you should do your operations.
JAX is definitely easier to do things that aren't on rails.
You could do the same with tensorflow and pytorch, but in my experience, with more difficulty since they're more opinionated about how you should do your operations.
JAX is definitely easier to do things that aren't on rails.