# The Python stack of wai-neural-e2e.yml (wai_validate_synthetic.py), as wanted. CI installs
# the hashed closure, requirements-neural-e2e.txt, generated from this file by its header's
# command. torch and torchvision come from the CPU wheel index, so their compiled ops
# match (compressai imports torchvision) and Linux does not resolve the CUDA builds.
# torch and torchvision: the CPU builds, pinned by torch-cpu-overrides.txt.
compressai
# `onnx` (the exporter) is separate from `onnxruntime` (the decoder).
onnx
onnxruntime
numpy
pillow>=13.0.0
# onnx's reference implementation uses ml_dtypes.float4_e2m1fn, which 0.5 added; without
# this floor the universal resolution settles on 0.4.1 and the exporter fails at import.
ml-dtypes>=0.5.0