simple-jax-to-tensorrt-example A MNIST model is defined and trained in Jax, then converted to TensorRT for inference.