backoff
ray[data,serve]>=2.49.0
ffmpeg-python
pandas<3,>=2.0
sqlglot>=30.0.0
tqdm>=4.66.0
typer>=0.12.0
click>=8.2.0
pyyaml>=6.0
fastapi>=0.114.0
uvicorn[standard]>=0.30.0
python-multipart>=0.0.9
prometheus-fastapi-instrumentator<8,>=7.0
opentelemetry-api>=1.41.1
opentelemetry-sdk>=1.41.1
opentelemetry-exporter-otlp-proto-grpc>=1.41.1
fastmcp>=2.0.0
httpx>=0.27.0
requests>=2.32.5
urllib3==2.7.0
pydantic>=2.8.0
rich>=13.7.0
universal-pathlib>=0.2.0
numpy>=1.26.0
debugpy>=1.8.0
fsspec>=2025.5.1
s3fs>=2025.5.1
pypdfium2==4.30.0
pillow==12.2.0
nltk==3.9.4
markitdown
langchain-nvidia-ai-endpoints>=1.4.0
lancedb
nvidia-riva-client>=2.25.1

[all]
nemo_retriever[benchmarks,llm,local,multimedia,nemotron-parse,service,tabular]

[benchmarks]
datasets>=4.8.2
open-clip-torch==3.2.0

[dev]
build>=1.2.2
pytest>=8.0.2

[llm]
litellm<2,>=1.86.0

[local]
glom
transformers<5,>=4.57.6
tokenizers>=0.21.1
accelerate==1.12.0
opencv-python-headless>=4.8.0
tritonclient
einops
easydict
addict
scikit-learn>=1.6.0
timm==1.0.22
albumentations==2.0.8
nemotron-page-elements-v3<4,>=3.0.1
nemotron-graphic-elements-v1<2,>=1.0.0
nemotron-table-structure-v1<2,>=1.0.0
nvidia-ml-py
apscheduler>=3.10
psutil>=5.9.0

[local:sys_platform == "darwin"]
torch==2.11.0
torchvision==0.26.0

[local:sys_platform == "linux"]
torch==2.11.0
torchvision==0.26.0
vllm==0.20.0
flashinfer-cubin==0.6.8.post1
flashinfer-python==0.6.8.post1

[local:sys_platform == "linux" and (platform_machine == "x86_64" or platform_machine == "aarch64")]
nemotron-ocr<3,>=2.0.0

[local:sys_platform == "win32"]
torch==2.11.0
torchvision==0.26.0

[multimedia]
soundfile>=0.12.0
scipy>=1.11.0
cairosvg>=2.7.0
librosa>=0.10.2
numba>=0.59

[nemotron-parse]
open-clip-torch==3.2.0

[service]
backoff
litellm<2,>=1.86.0
glom
easydict
addict
scikit-learn>=1.6.0
psutil>=5.9.0
apscheduler>=3.10
librosa>=0.10.2
numba>=0.59

[tabular]
duckdb>=1.2.0
duckdb-engine>=0.13.0
neo4j>=5.0
langgraph>=1.2.0
