{"library":"megatron-core","install":[{"cmd":"pip install megatron-core","imports":["from megatron.core.tensor_parallel.layers import ColumnParallelLinear","from megatron.core.tensor_parallel.layers import RowParallelLinear","from megatron.core.transformer.transformer_block import TransformerBlock","from megatron.core.transformer.transformer_layer import TransformerLayer","from megatron.core import dist_init"]},{"cmd":"pip install 'megatron-core[cuda]' # For full CUDA/cuDNN integration if needed beyond base","imports":[]}]}