Yep it just uses Jax to do the inference, no tensorflow needed.