new nn wip

NN/requirements.txt

@@ -1,13 +1,22 @@
-tensorflow>=2.5.0
+# Main dependencies
 numpy>=1.19.5
 pandas>=1.3.0
 matplotlib>=3.4.2
 scikit-learn>=0.24.2
-tensorflow-addons>=0.13.0
-plotly>=5.1.0
-h5py>=3.1.0
-tqdm>=4.61.1
-pyyaml>=5.4.1
-tensorboard>=2.5.0
-ccxt>=1.50.0
-requests>=2.25.1 
+
+# PyTorch (primary framework)
+torch
+torchvision
+
+# TensorFlow (optional)
+# tensorflow>=2.5.0
+# tensorflow-addons>=0.13.0
+
+# Additional dependencies
+plotly
+h5py
+tqdm
+pyyaml
+tensorboard
+ccxt
+requests