--index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://pypi.org/simple

##############################
# Core ML stack (CUDA build) #
##############################

torch==2.5.0
torchvision==0.20.0
torchaudio==2.5.0

-f https://data.pyg.org/whl/torch-2.5.0+cu121.html
torch_geometric
pyg_lib
torch_scatter
torch_sparse
torch_cluster
torch_spline_conv

#########################
# Numerics / science    #
#########################

numpy
scipy
pandas
scikit-learn

#########################
# Plotting & misc       #
#########################

matplotlib
seaborn
imageio
meshio
h5py
tensorboard

#########################
# Model / config helpers#
#########################

timm
einops
omegaconf
accelerate
huggingface-hub
lion-pytorch

#########################
# 3D / VTK / Trame stack#
#########################

vtk
pyvista

trame==3.12.0
trame-vtk==2.10.0
trame-vuetify==3.1.0
trame-server==3.6.2
trame-client==3.10.4
wslink

#########################
# Web / misc            #
#########################

pathlib2>=2.3.0
typing-extensions>=4.0.0
transformers>=4.20.0
trimesh
pytest
requests
PyYAML