ml_platform/pixi.toml at main · maniaclab/ml_platform · 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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
[workspace]
name = "ml-platform"
version = "2026.8.17"
description = "MaNIAC Lab ML platform environment"
channels = ["conda-forge"]
platforms = [
  "linux-64",
  "linux-aarch64",
  "osx-64",
  "osx-arm64",
  { name = "linux-64-cuda", platform = "linux-64", cuda = "13.0" },
]
requires-pixi = ">=0.76.2"  # https://github.com/maniaclab/ml_platform/issues/8
exclude-newer = "7d"

[feature.mlbase]
platforms = ["linux-64"]

[feature.mlbase.dependencies]

# Python
python = "==3.12"
pixi-kernel = "*"

# CERN ROOT
root = "*"

# Java (OpenJDK 8 in ml_base)
openjdk = "*"

# ML frameworks
gymnasium = "*"
keras = "*"
keras-tuner = "*"
tensorflow-datasets = "*"

# Data science
cython = "*"
h5py = "*"
numpy = "*"
pandas = "*"
pyarrow = "*"
pytables = "*"
scikit-learn = "*"
scipy = "*"

# Visualization
bokeh = "*"
graphviz = "*"
matplotlib = "*"
seaborn = "*"

# Jupyter ecosystem
dask-labextension = "*"
ipykernel = "*"
ipywidgets = "*"
jupyter = "*"
jupyter_bokeh = "*"
jupyter-collaboration = "*"
jupyter_client = ">=8.6.3"
jupyterhub-singleuser = ">=5.4.2,<6"
jupyterlab = "*"
jupyterlab-git = "*"
jupyterlab_server = "*"
metakernel = "*"
pycrdt = "*"
rise = "*"

# HEP
uproot = "*"

# System tools
compilers = "*"
curl = "*"
git = "*"
jq = "*"
make = "*"
pkg-config = "*"
rsync = "*"
unzip = "*"
vim = "*"
wget = "*"
zip = "*"

# Libraries
freetype = "*"
hdf5 = "*"
libpng = "*"
tbb = "*"
zeromq = "*"
zlib = "*"

# Other
elasticsearch = "*"
imageio = "*"
plumbum = "*"
pillow = "*"
qtpy = "*"
requests = "*"
tqdm = "*"

[feature.mlbase.pypi-dependencies]
# Packages NOT on conda-forge (or easier from PyPI)
jupyter-mcp-server = ">=2.2.3,<3"
jupyter-mcp-tools = ">=0.1.8"

[feature.mlcpu.dependencies]
tensorflow = "*"

[feature.mlgpu]
platforms = ["linux-64-cuda"]

[feature.mlgpu.dependencies]
tensorflow-gpu = "*"

[feature.dev.dependencies]
python = "==3.11"
tbump = "*"

[feature.dev.tasks]
tbump = { cmd = "tbump", description = "Run tbump" }
bump = { cmd = "tbump $(date +\"%Y.%-m.%-d\")", description = "Run tbump with current date" }

[environments]
dev = ["dev"]
mlcpu = ["mlbase", "mlcpu"]
ml = ["mlbase", "mlgpu"]