czy ktoś ma doświadczenie w trenowaniu (na dużą skalę) zarówno w Jax, jak i Pytorch? jeśli tak, to na jaką skalę naprawdę odczuwasz ból w Pytorch, a na jaką skalę czujesz, że ogólny ból Jax jest zdecydowanie wart tego?