IPython
jaxtyping
matplotlib
numpy
pandas
seaborn
scipy
tenacity
tensorboard
torch
torchaudio
torchvision
einops
lightning
wandb
hydra-core
omegaconf
h5py
opencv-python-headless
imageio>=2.36
moviepy>=2.0.0
huggingface_hub>=0.25
jupyter
ipywidgets

[dev]
ruff>=0.9
mypy>=1.0
pytest
types-psutil
types-tqdm
types-PyYAML
types-requests
nbmake
jupyterlab

[devmodels]
neuralpredictors
