does anyone have experience training (at scale) in both Jax and Pytorch? if so, at what scale do you really find the pain hits in Pytorch and at what scale do you feel like the general pain of Jax is definitively worth it?
22,08K