are cineva experiență de antrenament (la scară) atât în Jax, cât și în Pytorch? dacă da, la ce scară găsești cu adevărat loviturile durerii în Pytorch și la ce scară simți că durerea generală a lui Jax merită cu siguranță?