I agree with your point that people hours are worth more than GPU hours (except for large-scale training runs or inference workloads of course…). Having been developing in all three frameworks (torch, TF, JAX) extensively over the past year, I've personally found JAX to be the

→ View original post on X — @fchollet,