© 2025-2026 PySpect
| Dependency | Optional | Group |
|---|---|---|
| transformer-engine-cu12 | true | core |
| transformer-engine-cu13 | true | core-cu13 |
| transformer-engine-jax | true | jax |
| transformer-engine-torch | true |
| Package | Optional | Group |
|---|---|---|
| megatron-bridge | true | te |
| megatron-core | true | te |
| nvidia-physicsnemo | true | perf |
| pytorch |