vigneshwar234 commited on
Commit
836634c
·
verified ·
1 Parent(s): 5038ae1

Add requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +27 -0
requirements.txt ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core
2
+ torch>=2.2.0
3
+ torchvision
4
+ torchaudio
5
+
6
+ # TMT-specific
7
+ einops>=0.7.0
8
+ torch-geometric
9
+
10
+ # NLP / data
11
+ transformers>=4.40.0
12
+ datasets>=2.19.0
13
+ tokenizers>=0.19.0
14
+
15
+ # Experiment tooling
16
+ wandb>=0.17.0
17
+ tensorboard>=2.16.0
18
+ torchviz
19
+ matplotlib>=3.8.0
20
+ jupyter>=1.0.0
21
+ pandas>=2.2.0
22
+ tabulate>=0.9.0
23
+
24
+ # Dev
25
+ pytest>=8.0.0
26
+ black>=24.0.0
27
+ ruff>=0.4.0