---
# Fast Conda Environment for ML
# Optimized with mamba for data scientists
name: ml_env
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  # Python
  - python=3.10
  # ML Frameworks (conda-optimized)
  - pytorch>=1.9.0
  - torchvision>=0.10.0
  - numpy>=1.21.0
  - pandas>=1.3.0
  - scikit-learn>=1.0.0
  - xgboost>=1.5.0
  # Data Science Tools
  - matplotlib>=3.5.0
  - seaborn>=0.11.0
  - jupyter>=1.0.0
  - notebook>=6.4.0
  - ipykernel>=6.0.0
  # Development Tools
  - pip
  - setuptools
  - wheel
  # GPU Support (if available)
  - cudatoolkit=11.3
  - pytorch-cuda>=11.3
  # pip fallback packages (if conda doesn't have them)
  - pip:
      - tensorflow>=2.8.0
      - statsmodels>=0.13.0
      - plotly>=5.0.0
      - dash>=2.0.0
      # ML Experiment Tracking
      - mlflow>=2.0.0
      - wandb>=0.13.0
      # Dashboard & Visualization
      - streamlit>=1.20.0
      - panel>=1.0.0
      - bokeh>=3.0.0
      # Data Science Tools
      - dvc>=3.0.0
      - optuna>=3.0.0
      - hyperopt>=0.2.0