zea==0.0.4 jax[cuda12] tyro optuna gradio spaces