jax
9f5f6edb - [Pallas] Fix integer array indexing

Commit

305 days ago

[Pallas] Fix integer array indexing Fixes https://github.com/google/jax/issues/22783 jax-fixit PiperOrigin-RevId: 746260869

References

Author

ayaka14732

Committer

Google-ML-Automation

Parents