jax icon indicating copy to clipboard operation
jax copied to clipboard
verified publisher icon google

[Pallas] Fix integer array indexing

Open ayaka14732 opened this issue 1 year ago • 1 comments

Fixes https://github.com/google/jax/issues/22783

Note: the code works now but still need some clean up

ayaka14732 avatar Sep 03 '24 18:09 ayaka14732

Blocked by https://github.com/google/jax/pull/23534

ayaka14732 avatar Sep 10 '24 16:09 ayaka14732

Closing due to merge conflict caused by https://github.com/google/jax/pull/23583. Will make another PR.

ayaka14732 avatar Sep 19 '24 11:09 ayaka14732

Superseded by https://github.com/google/jax/pull/23758

ayaka14732 avatar Sep 19 '24 12:09 ayaka14732