llvm/mlir/python/requirements.txt

numpy>=1.19.5, <=1.26
pybind11>=2.9.0, <=2.10.3
PyYAML>=5.4.0, <=6.0.1
ml_dtypes>=0.1.0, <=0.4.0   # provides several NumPy dtype extensions, including the bf16