jaxtyping>=0.2.25
torch
tqdm

[all]
bayesian_lora[examples]
bayesian_lora[dev]
bayesian_lora[docs]

[dev]
ipywidgets>=8.0.4
jupyterlab<3.6,>=3.5
jupyterlab-vim
jupyterlab-vimrc
mypy<=1.0,>=0.990
tensorboard<3.0,>=2.11.2

[docs]
furo>=2022.9.29
sphinx-autobuild>=2021.3.14
sphinx-copybutton>=0.5.1
sphinxext-opengraph>=0.7.2

[examples]
datasets>=2.16.1
hydra-core<2.0,>=1.2.0
omegaconf>=2.3.0
peft>=0.5.0
torchmetrics>=1.2.0
transformers>=4.37.2
pytest>=7.2.0
