diff --git a/requirements.txt b/requirements.txt
index 3446439..967f3e3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,50 +1,12 @@
-aiohappyeyeballs~=2.6.1
-aiohttp~=3.12.12
-aiosignal~=1.3.2
-attrs~=25.3.0
-black~=25.1.0
-click~=8.2.1
-filelock~=3.18.0
-frozenlist~=1.7.0
-fsspec~=2025.5.1
-h5py~=3.14.0
-idna~=3.10
-Jinja2~=3.1.6
-lightning~=2.5.1.post0
-lightning-utilities~=0.14.3
-MarkupSafe~=3.0.2
-mpmath~=1.3.0
-multidict~=6.4.4
-mypy_extensions~=1.1.0
-networkx~=3.4
-numpy~=2.2.6
-nvidia-cublas-cu12~=12.6.4.1
-nvidia-cuda-cupti-cu12~=12.6.80
-nvidia-cuda-nvrtc-cu12~=12.6.77
-nvidia-cuda-runtime-cu12~=12.6.77
-nvidia-cudnn-cu12~=9.5.1.17
-nvidia-cufft-cu12~=11.3.0.4
-nvidia-cufile-cu12~=1.11.1.6
-nvidia-curand-cu12~=10.3.7.77
-nvidia-cusolver-cu12~=11.7.1.2
-nvidia-cusparse-cu12~=12.5.4.2
-nvidia-cusparselt-cu12~=0.6.3
-nvidia-nccl-cu12~=2.26.2
-nvidia-nvjitlink-cu12~=12.6.85
-nvidia-nvtx-cu12~=12.6.77
-packaging~=24.2
-pathspec~=0.12.1
-platformdirs~=4.3.8
-propcache~=0.3.2
-pytorch-lightning~=2.5.1.post0
-PyYAML~=6.0.2
-setuptools~=80.9.0
-sympy~=1.14.0
-torch~=2.7.1
-torchmetrics~=1.7.3
-tqdm~=4.67.1
-triton~=3.3.1
-typing_extensions~=4.14.0
-yarl~=1.20.1
+h5py
+pytorch-lightning
+matplotlib
+numpy
+PyYAML
+scikit-learn
+timm
+torch
+onnx
+onnxruntime
 --index-url https://git.riahub.ai/api/packages/qoherent/pypi/simple/ 
 utils
\ No newline at end of file