PySpect
Home
Lists
Top stars
Rising stars
← Back to transformer-engine-jax
Dependencies for transformer-engine-jax
Frequently asked questions
© 2025-2026 PySpect
Dependencies of transformer-engine-jax (4).
Dependency
Optional
Group
flax
false
jax
false
nvidia-cudnn-frontend
false
transformer-engine-cu12
false
Dependent packages
Packages depending on transformer-engine-jax (1).
Package
Optional
Group
transformer-engine
true
jax