docker.io/javirub/flashattention-pytorch
该Docker镜像包含PyTorch环境以及FlashAttention库，可用于在PyTorch框架下进行高效的注意力机制相关的深度学习模型训练与推理任务，简化了FlashAttention的部署与使用流程。