ninja  # For faster builds.
psutil
pandas  # Required for Ray data.
pyarrow  # Required for Ray data.
sentencepiece  # Required for LLaMA tokenizer.
numpy
torch ~= 2.0.0 # GKE doesn't have the latest CUDA drivers
transformers >= 4.33.1  # Required for Code Llama.
xformers >= 0.0.22
fastapi
uvicorn[standard]
ai2-olmo # FIXME: This is specific to OLMo
