torch
transformers
mistral_inference