requirements.txt

-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax[cuda12_pip]
jaxlib==0.4.7+cuda12.cudnn88
flax==0.6.8
fastapi==0.95.0
uvicorn[standard]==0.21.1
Pillow==9.4.0
tensorflow
tensorflow-datasets
python-jose[cryptography]
passlib[bcrypt]