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[cpu]>=0.4.16  # FEATURE=[lite]
jaxlib>=0.4.16  # FEATURE=[lite]
kuscia==0.0.2.dev240320  # FEATURE=[lite]
multiprocess  # FEATURE=[lite]
networkx==2.8.8
numba==0.57.0  # FEATURE=[lite]
numpy  # FEATURE=[lite]
# TODO: onnx and tensorflow have dependency conflicts, waiting for upstream to resolve
# onnx==1.13.1
pandas==1.5.3  # FEATURE=[lite]
tabulate # FEATURE=[lite]
# For security reason.
protobuf>=4,<5 # FEATURE=[lite]
pyarrow==14.0.2  # FEATURE=[lite]
s3fs==2024.2.0 # FEATURE=[lite]
scikit-learn==1.3.2  # FEATURE=[lite]
sdc-apis==0.1.0.dev240320; platform_machine == "x86_64"  # FEATURE=[lite]
sdc-sdk==0.1.0.dev240320; platform_machine == "x86_64"  # FEATURE=[lite]
ray==2.9.1  # FEATURE=[lite]
secretflow-rayfed==0.2.1a1 # 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.9.0.dev20240321 # FEATURE=[lite]
sf-heu==0.5.0.b0 # FEATURE=[lite]
secretflow_serving_lib==0.3.0.dev20240320 # FEATURE=[lite]
tensorflow-macos>=2.12,<2.13; platform_system == "Darwin"
tensorflow>=2.12,<2.13; platform_system != "Darwin"
tf2onnx>=1.13.0
tqdm
validator.py==1.3.0
xgboost==1.7.5
# For security reason.
wheel>=0.38.1
torch==2.1.2
torchmetrics==1.3.2
torchvision==0.16.2
torchaudio==2.1.2
interconnection==0.2.0.dev0 # FEATURE=[lite]