[vlovich123]

What inference framework is this using to accelerate the math? I only found references to numpy and Jax but this doesn’t seem to use tensorflow?

[sakras]

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