Roy Frostig 753a6b2b0c update experimental.sparse to new-style typed keys

2024-03-07 12:40:10 -08:00

..

__init__.py

[sparse] implement bcsr_concatenate

2023-02-15 14:10:56 -08:00

_base.py

Replaced most usages of abc.ABC with util.StrictABC

2024-01-29 12:40:43 +00:00

_lowerings.py

[ROCm]: Lower sparse(some) ops correctly for ROCm

2023-12-18 20:47:44 +00:00

ad.py

typing: fix incorrect tuple annotations

2024-02-26 10:53:19 -08:00

api.py

Upgrade remaining sources to Python 3.9

2023-12-13 10:29:45 +00:00

bcoo.py

jnp.unique: make return_inverse shape match NumPy 2.0

2024-01-16 11:47:45 -08:00

bcsr.py

Integrate StableHLO at openxla/stablehlo@ab709fe4

2023-12-11 12:30:50 -08:00

coo.py

Integrate StableHLO at openxla/stablehlo@ab709fe4

2023-12-11 12:30:50 -08:00

csr.py

Fix duplicate word occurrences

2023-12-19 06:15:30 +01:00

linalg.py

Fix duplicate word occurrences

2023-12-19 06:15:30 +01:00

random.py

update experimental.sparse to new-style typed keys

2024-03-07 12:40:10 -08:00

test_util.py

Upgrade remaining sources to Python 3.9

2023-12-13 10:29:45 +00:00

transform.py

update experimental.sparse to new-style typed keys

2024-03-07 12:40:10 -08:00

util.py

[sparse] fix concretization error for nse

2023-11-06 14:03:16 -08:00