requirements.txt

# Avoid thead pool fork issue, see https://github.com/grpc/grpc/issues/31772
grpcio<=1.49.1
jax
jaxlib
multiprocess
networkx==2.8.8
numpy==1.23.5
# TODO: onnx and tensorflow have dependency conflicts, waiting for upstream to resolve
# onnx==1.13.1
pandas==1.5.3
# For security reason.
protobuf==3.19.6
pyarrow==11.0.0
s3fs==2022.1.0
scikit-learn==1.1.3
sdc-apis==0.1.0b0; platform_system != "Darwin"
sdc-sdk==0.1.0b0; platform_system != "Darwin"
secretflow-rayfed==0.1.1a3
# For security reason. This is introduced by TensorFlow.
setuptools>=65.5.1
sparse>=0.14.0
spu==0.3.3b2
sf-heu==0.4.3b3
tensorflow-macos==2.11.0; platform_machine == "arm64" and platform_system == "Darwin"
tensorflow==2.11.1; platform_machine != "arm64"
tf2onnx>=1.13.0
tqdm
validator.py==1.3.0
xgboost==1.7.5
# For security reason.
wheel>=0.38.1
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.0.0
torchmetrics==0.11.4
torchvision==0.15.1
torchaudio==2.0.1
click