jax_ai_stack
