work vijax A simple JAX library to easily run VI vistan A simple Python library to run VI on Stan models