у кого-нибудь есть опыт обучения (в большом масштабе) как в Jax, так и в Pytorch? Если да, то на каком масштабе вы действительно ощущаете трудности в Pytorch и на каком масштабе вы чувствуете, что общие трудности Jax определенно стоят того?