torch
packaging
ninja
einops
triton
transformers
