requirements.txt

# Core ML Frameworks
# PyTorch: Installiere mit CUDA 12.8 support via:
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
torch>=2.6.0
torchvision>=0.21.0
torchaudio>=2.6.0
transformers>=4.50.0
accelerate>=0.30.0

# Profiling & Metriken
pynvml>=11.5.0

# Quantization Backends
hqq>=0.2.0
optimum-quanto>=0.0.16

# Attention Backends
sageattention>=2.2.0

# Benchmarks
lm-eval>=0.4.8
datasets>=2.20.0

# Datenverarbeitung
numpy>=1.24
pandas>=2.1
scipy>=1.11

# Visualisierung
matplotlib>=3.8
seaborn>=0.13
plotly>=5.18

# Experiment-Tracking (optional)
# wandb>=0.16
# mlflow>=2.8

# Utilities
tqdm>=4.66
pyyaml>=6.0
jsonlines>=4.0

# Development
ipython==8.18.1
jupyter==1.0.0
black==23.12.0
flake8==6.1.0

# Optional: vLLM (schnelle Inferenz)
# vllm==0.2.5  # Benötigt CUDA 12.x

# Optional: TensorRT-LLM
# tensorrt-llm==0.5.0

# Note: Install PyTorch FIRST with CUDA support:
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
# Then install rest: pip install -r requirements.txt