transformer_engine[pytorch]>=2.0.0
accelerate>=1.10.0