Working with google's JAX inside a gunicorn/flask server

Question

I want to serve an application that processes data within the googles JAX framwork with flask and gunicorn.

If run inside flask, everything works fine. As soon as I r