Deep Learning

i keep hitting this in Deep Learning and the docs skip the failure path. is the miss supposed to throw or just return null?

currently benchmarking a medium scale transformer training loop. pytorch is hitting a wall with compile overhead on complex custom kernels. seeing some people claim jax is faster, but worried the migration cost outweighs the gain. anyone actually ship this in prod recently?