IPython
jaxtyping
matplotlib
numpy
pandas
standard-imghdr
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
mlflow

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

[devmodels]
neuralpredictors

[optuna]
hydra-optuna-sweeper
