vlm-gym inference TUI

Small vision model in JAX for geoguesssing