ringattention
ringattention copied to clipboard
Incorrect project requirements
In the project requirements, it is specified that the version of jax is 0.4.13. However, Pallas was added in the version 0.4.16 (https://github.com/google/jax/commit/d872812a359a3bafcfdeba1fcdb874ec77c209db).