requirements.txt

# PyTorch
torch==2.5.1
torchvision==0.20.1
av==14.0.1 # required for torchvision video
torchaudio==2.5.1
torchdata==0.9.0
torchinfo==1.8.0
torchmetrics==1.6.0
# Anciliary PyTorch Lightning - since 2.0 it got less useful
lightning==2.4.0
# HuggingFace
datasets[vision,audio]==3.1.0
huggingface-hub[torch, cli]==0.26.5
timm==1.0.12
# network quantization
brevitas==0.11.0 # conflict with NumPy > 2.0
# for hyperdimensional computing
torch-hd==5.7.1

# Data wrangning
imageio==2.36.1
numpy==1.26.4
pandas==2.2.3
scipy==1.14.1
opencv-contrib-python-headless==4.10.0.84
einops==0.8.0

# GPU accelerated procesing
#cupy-cuda12x==13.3.0
#--extra-index-url=https://pypi.nvidia.com
#cudf-cu12==24.10.*
#cuml-cu12==24.10.*

# Classic learning
scikit-learn==1.6.0
sympy==1.13.1

# Visualization
umap-learn[plot]==0.5.7
matplotlib==3.9.2
seaborn==0.13.2
plotly==5.24.1
kaleido==0.2.1

# Storage
h5py==3.12.1
lz4==4.3.3

# Notebooks
ipykernel==6.29.5
ipywidgets==8.1.5
nbformat==5.10.4
nbconvert==7.16.4
papermill==2.6.0

# Logging
comet-ml==3.47.4

# Anciliary
line-profiler==4.2.0

# Extra 
mediapipe==0.10.18

# Events
aedat==2.1.0
# dv==1.0.12
# dv-processing==1.7.9

# Hidden Markov Models
hmmlearn==0.3.3 # conflict with NumPy > 2.0
pomegranate==1.1.1 # much faster HMM

# Dev utils
pip-tools==7.4.1