umchand/rocm_pytorch_flash_attn

•Updated over 1 year ago

rocm pytorch with flash attention

Image

878

umchand/rocm_pytorch_flash_attn repository overview

No overview available

This repository doesn't have an overview

Recent tags

Content type

Image

Digest

sha256:d3158163f…

Size

21.9 GB

Last updated

over 1 year ago

docker pull umchand/rocm_pytorch_flash_attn