v0.4.1

I think we can just use pytorch's builder image, which seems trustwor…

v0.3.0

Merge branch 'cmake'.

Switches the build system to cmake, which is more flexible, and fixes up the compatibility for jax 0.5.0--0.7.0.

v0.3.0a

Clean up imports in flash.py, flash_sharding.py, and ring_attention.py.

v0.2.2

Build cuda12 version with Hopper support.

v0.2.1

Bump minor version to 0.2.1.

v0.2.0

Expanded vmap support for flash_mha. Vmapping q but not k,v reduces t…

v0.1.0a3

Try cibuildwheel.

v0.1.0a2

Try cibuildwheel.

v0.1.0a1

Set CUDA_HOME for the sdist

v0.1.0

Try release with github actions and rebased repo.