| # BioRLHF Requirements | |
| # Install with: pip install -r requirements.txt | |
| # Core ML | |
| torch>=2.0.0 | |
| transformers>=4.36.0 | |
| datasets>=2.14.0 | |
| accelerate>=0.24.0 | |
| peft>=0.6.0 | |
| trl>=0.7.0 | |
| # Quantization | |
| bitsandbytes>=0.41.0 | |
| # Logging | |
| wandb>=0.15.0 | |
| # Scientific | |
| pandas>=2.0.0 | |
| numpy>=1.24.0 | |
| scipy>=1.10.0 | |
| scikit-learn>=1.3.0 | |
| # Utilities | |
| tqdm>=4.65.0 | |
| jsonlines>=3.1.0 | |
| # Optional: Flash Attention (install separately) | |
| # pip install flash-attn --no-build-isolation | |