how to install this lib by pip

nshepperd / flash_attn_jax

JAX bindings for Flash Attention v2

BSD 3-Clause "New" or "Revised" License

62 stars 0 forks source link

how to install this lib by pip #1

Closed pass-lin closed 3 months ago

nshepperd commented 3 months ago

At the moment you need to install directly from one of the wheels in the release page. I'm planning to upload to pypi at some point but not sure how to do versioning so i haven't yet.