simple-jax-to-tensorrt-example

A MNIST model is defined and trained in Jax, then converted to TensorRT for inference.