Can I write terminal-based games in JAX?

[rugvedpund]

I am looking to make a physics-based game based on a simulation. Let's say it is simple enough that it can be played in plain text on the terminal.

My question is regarding the best way to package the all the JIT dependencies into a nice executable. In particular, to be playable on machines that may or may not have a GPU.

While there is sufficient documentation (which I still need to learn) to convert pure python packages to executables, I would appreciate some guidance if I were to JAX to simplify the engine.