{"library":"transformer-engine","install":[{"cmd":"pip install transformer-engine","imports":["import transformer_engine.pytorch as te\nlinear_layer = te.Linear(in_features, out_features)","import transformer_engine.pytorch as te\nlayer_norm = te.LayerNorm(normalized_shape)","from transformer_engine.pytorch import fp8_autocast\nwith fp8_autocast():\n    output = model(input)","import transformer_engine.pytorch as te\ntransformer_block = te.TransformerLayer(...) "]}]}