{"library":"transformer-engine-cu12","install":[{"cmd":"pip install --no-build-isolation transformer-engine-cu12[pytorch]","imports":["from transformer_engine.pytorch import Linear","from transformer_engine.pytorch import LayerNorm","from transformer_engine.pytorch import TransformerLayer","from transformer_engine.pytorch.fp8 import fp8_autocast"]},{"cmd":"pip install --no-build-isolation transformer-engine-cu12[jax]","imports":[]},{"cmd":"pip install --no-build-isolation transformer-engine-cu12[core]","imports":[]}]}