-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jax[cuda11_pip] jaxlib git+https://github.com/huggingface/diffusers@main opencv-python transformers flax pillow torch torchvision accelerate