requirements.txt

#
# FEATURE=[lite] is a specific comment, indicating that this dependency will be
# used as a dependency of secretflow-lite.
#
# Avoid thead pool fork issue, see https://github.com/grpc/grpc/issues/31772
clean-text  # FEATURE=[lite]
click  # FEATURE=[lite]
grpcio==1.56.2   # FEATURE=[lite]
jax<=0.4.12  # FEATURE=[lite]
jaxlib<=0.4.12  # FEATURE=[lite]
kuscia==0.0.2.dev231025  # FEATURE=[lite]
multiprocess  # FEATURE=[lite]
networkx==2.8.8
numba==0.57.0  # FEATURE=[lite]
numpy==1.23.5  # FEATURE=[lite]
# TODO: onnx and tensorflow have dependency conflicts, waiting for upstream to resolve
# onnx==1.13.1
pandas==1.5.3  # FEATURE=[lite]
# For security reason.
protobuf==3.19.6  # FEATURE=[lite]
pyarrow==14.0.2  # FEATURE=[lite]
s3fs==2022.1.0
scikit-learn==1.3.2  # FEATURE=[lite]
sdc-apis==0.1.0b0; platform_system != "Darwin"  # FEATURE=[lite]
sdc-sdk==0.1.0b0; platform_system != "Darwin"  # FEATURE=[lite]
secretflow-rayfed==0.2.0a19 # FEATURE=[lite]
# For security reason.
setuptools>=65.5.1  # FEATURE=[lite]
# For security reason.
certifi>=2023.7.22  # FEATURE=[lite]
sparse>=0.14.0
spu==0.8.0.dev20240112  # FEATURE=[lite]
sf-heu==0.5.0b0  # FEATURE=[lite]
secretflow_serving_lib==0.2.0.dev20240129 # FEATURE=[lite]
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
torch==2.1.1
torchmetrics==0.11.4
torchvision==0.16.1
torchaudio==2.1.1
interconnection==0.1.0b1 # FEATURE=[lite]