I have never used TRC, but I heard from many that if you reach out to them, they are really helpful. I have not converted to TPU, because it is literally offer

I have never used TRC, but I heard from many that if you reach out to them, they are really helpful. I have not converted to TPU, because it is literally offered by only one company. It will be the height of "vendor lock-in". But, I must say that JAX on TPU is faster than anything and everything that I have ever seen.