keras jax[cuda12] pydot matplotlib pandas tqdm scikit-learn