ai-research-env/environment.yml at main · eotles/ai-research-env · 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
name: ai-research-env
channels:
  - pytorch
  - conda-forge
  - defaults
channel_priority: strict
dependencies:
  - python=3.11
  - pip

  # core scientific python
  - numpy
  - pandas
  - scipy
  - scikit-learn

  # visualization and notebooks
  - matplotlib
  - jupyterlab
  - ipykernel
  - ipywidgets

  # quality-of-life (progress, logging, utilities)
  - tqdm
  - rich
  - pyyaml
  - pydantic
  - python-dotenv
  - requests

  # data I/O and local analytics
  - pyarrow
  - duckdb
  - polars
  - h5py

  # performance
  - numba

  # dev tooling
  - pytest
  - pytest-cov
  - ruff
  - mypy
  - pre-commit
  - typer
  - joblib

  # experiment tracking
  # - mlflow

  # pytorch stack (from pytorch channel, consistent across platforms)
  - pytorch
  - torchvision
  - torchaudio
  - cpuonly

  # tensorflow (pin to a version that is available across channels/platforms)
  - tensorflow=2.18.*

  - jupyterlab
  - ipykernel