This repository was archived by the owner on Nov 17, 2023. It is now read-only.

This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Can We implement Flash attention 2 in MXnet #21222

Open

Labels

Feature request

opened

Description

Flash Attention 2 is a library that provides attention operation kernels for faster and more memory efficient inference and training:

References

list known implementations

to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

No one assigned

Labels

Feature request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests