search

iree-org / iree-jax

Apache License 2.0
48 stars 19 forks source link

Update to jax/jaxlib 0.4.2 #48

Closed rsuderman closed 1 year ago

rsuderman commented 1 year ago

There were some fixes required to update to jax 0.4.2. This includes a change to GlobalOp initialization.