torch>=2.0
triton>=3.4.0

[dev]
pytest

[notebook]
pylate==1.3.4
sentence-transformers
jupyter
matplotlib
