# Direct dependencies for TEDBench (Python 3.10, PyTorch 2.11)
# Install with: uv pip install -r requirements.txt

# Core ML
torch
numpy
scipy
scikit-learn
torchmetrics

# Training framework
lightning
hydra-core

# Protein / structure
fair-esm
biotite
biopython

# Transformers / tokenization
transformers
huggingface_hub
datasets

# Data / serialization
cloudpathlib
msgpack_numpy
zstd
brotli

# Utilities
einops
joblib
rich
tqdm
matplotlib
