Sign inSign up

rocm/xformers

Verified Publisher

By AMD

Updated almost 2 years ago

Image
1

8.4K

rocm/xformers repository overview

Prebuilt images for xformers with Composable Kernel Flash Attention extensions (AMD MI Instinct GPUs only)

Pytorch: 10/01/2024 nightly ROCm: 6.2.2 Python: 3.11

Tag summary

Content type

Image

Digest

sha256:6b0a9df50

Size

10.9 GB

Last updated

almost 2 years ago

docker pull rocm/xformers