Matthew Johnson 251b93ebd7 fixups that we meant to include in #26427 ...

Co-authored-by: Dougal Maclaurin <dougalm@google.com>

2025-03-08 00:03:26 +00:00

..

__init__.py

Cleanup: fix unused imports & mark exported names

2024-10-16 17:42:41 -07:00

discharge.py

Removed unused `# type: ignore` comments

2025-02-13 21:12:27 +00:00

indexing.py

[mutable-arrays] read values should have the same explicit sharding as ref

2025-03-07 20:53:29 +00:00

primitives.py

fixups that we meant to include in #26427

2025-03-08 00:03:26 +00:00

types.py

[mutable-arrays] read values should have the same explicit sharding as ref

2025-03-07 20:53:29 +00:00

utils.py

[better_errors] Continue adding debug info to Jaxprs (step 6)

2025-02-11 11:28:58 +01:00