torch
packaging
ninja
einops
transformers
