numpy
pyyaml

[all]
flash-rt[chameleon,eval,jax,minimax-remover,motus,server,torch]

[chameleon]
torch
safetensors
transformers<4.56,>=4.43
Pillow

[eval]
opencv-python
tqdm

[flash-attn]
flash-attn

[jax]
jax
jaxlib
ml_dtypes
orbax-checkpoint
flax

[minimax-remover]
diffusers
einops
scipy
sageattention

[motus]
nvidia-cublas<14,>=13.1

[server]
fastapi
uvicorn

[thor-fa4]
nvidia-cutlass-dsl==4.5.1
quack-kernels==0.4.1

[torch]
torch
safetensors
sentencepiece
