# Non Colab Packages
# fit dependencies
# Core ML/AI Framework
torch>=2.5.1
torchvision>=0.20.1
torchaudio>=2.5.1
datasets>=3.6.0
sentence-transformers==5.1.0
gpustat==1.1.1

# Distributed Computing
click<8.3.0
ray<=2.49.0

# LLM Inference
transformers>=4.56.1

# OpenAI API
openai==1.106.1
tiktoken==0.11.0

# LangChain Ecosystem
langchain
langchain-classic
langchain-core
langchain-community
langchain-openai
langchain-huggingface

# Vector Search
faiss-gpu-cu12==1.13.0

# Statistical Analysis
scipy>=1.16.1

# Data Manipulation & Display
pandas>=2.3.2
pyarrow==21.0.0
numpy>=1.26.4,<2.3
unstructured==0.18.15


# Notebook
ipykernel==6.30.1
ipywidgets>=7.3.4,<9.0.0  # Support both v7 (Colab) and v8 (Jupyter)
tensorboard>=2.11.0


# Other
psutil==7.0.0
tqdm==4.67.1
typing-extensions>=4.0.0
peft>=0.17.0
trl==0.21.0
bitsandbytes>=0.47.0
nltk>=3.9.1
evaluate>=0.4.5
rouge-score>=0.1.2
sentencepiece>=0.2.1
# mlflow>=3.2.0
requests>=2.32.0  # Relaxed for Colab (2.32.4)
loguru>=0.7.3
ipython>=7.34.0  # Colab compatibility (7.34.0)
jupyter>=1.1.1
