torch
tensorflow
tf-keras
transformers
transformers[torch]