torch numpy transformers==4.44.2 datasets==2.12.0 matplotlib tqdm openai nltk groq