Closed hadipash closed 4 months ago
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).
jax
0.4.13
0.4.16
Updated! Thanks!
In the project requirements, it is specified that the version of
jax
is0.4.13
. However, Pallas was added in the version0.4.16
(https://github.com/google/jax/commit/d872812a359a3bafcfdeba1fcdb874ec77c209db).