einops
colored
polygraphy
tensorrt==10.4.0
cuda-python