JAX vs TensorRT

[AastaLLL]

Hi,

Sorry that we don’t have too much experience on JAX.

But it seems that JAX is a language-level optimizer.
TensorRT is an algorithm-level optimizer and should give you a better performance.

Thanks.