Change Log 0.1.3 Changes Upgrade Jax from 0.4.30 to 0.4.33. 0.1.2 Changes Upgrade Python to 3.10 Fall back to triton backend for qkv in fp32 or with bias on gpu flash attention. 0.1.1 Changes Upgrade Jax from 0.4.28 to 0.4.30. 0.1.0 (Aug 22, 2024) Changes Add changelog.