torch
transformers
tensorflow>2.6