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 abo

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.