torch
transformer_engine[pytorch]
transformers
pytest