foundation-models-benchmark/environment.yml at main · MaxRonce/foundation-models-benchmark · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
name: fmb
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults

dependencies:
  # Core Python
  - python=3.12

  # Deep Learning & GPU
  - pytorch>=2.0
  - torchvision
  - pytorch-cuda=12.1

  # Scientific Computing
  - numpy
  - scipy
  - pandas

  # Visualization
  - matplotlib
  - seaborn

  # Machine Learning
  - scikit-learn

  # Data Loading
  - pyyaml
  - tqdm

  # CLI
  - pip

  # Pip-only packages
  - pip:
    # CLI framework
    - typer>=0.9

    # HuggingFace ecosystem
    - datasets
    - huggingface_hub
    - transformers
    - timm

    # Model utilities
    - einops

    # Astronomy
    - astropy

    # Normalizing Flows
    - FrEIA
    - normflows

    # Dimensionality reduction
    - umap-learn

    # Explainability
    - shap

    # Install FMB package
    - -e .