Running @jit functions continuously in real-time?

[bhchiang]

Don't think JAX is designed for this, but wondering if it'd be possible to use JAX in asynchronous / multi-threaded use cases?

For example, running KF in real-time on incoming data or a simple 3D software rasterizer?

Going from this:

@jit
def render( ... ):
   ...

while True:
   render( ... )

To running the function continuously in another thread. Any suggestions for best practices or links to code that do something similar would be greatly appreciated. Thanks!