Matthew Johnson 8a04dfd830 rolling back shard_map transposition change to fix a bug ...

Reverts 437d7be73534403f39fbee9d6391be1c532933a1

PiperOrigin-RevId: 561730581

2023-08-31 12:39:56 -07:00

..

control_flow

Remove the canonicalize_dtypes argument from mlir.ir_constant(s).

2023-08-17 06:44:12 -07:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

ann.py

Make mlir.custom_call() more general and expose it as jax.interpreters.mlir.custom_call().

2023-08-29 08:50:07 -07:00

convolution.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

eigh.py

Improve the bisection strategy in eigh.

2023-05-04 18:01:23 -07:00

fft.py

Remove the canonicalize_dtypes argument from mlir.ir_constant(s).

2023-08-17 06:44:12 -07:00

lax.py

rolling back shard_map transposition change to fix a bug

2023-08-31 12:39:56 -07:00

linalg.py

Make mlir.custom_call() more general and expose it as jax.interpreters.mlir.custom_call().

2023-08-29 08:50:07 -07:00

other.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

parallel.py

rolling back shard_map transposition change to fix a bug

2023-08-31 12:39:56 -07:00

qdwh.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

slicing.py

lax.dynamic_slice: avoid negative index correction for unsigned indices

2023-07-25 13:09:09 -07:00

special.py

Merge pull request #17144 from jakevdp:zeta

2023-08-18 11:04:43 -07:00

stack.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

svd.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

utils.py

Merge pull request #16419 from mattjj:pow-jvp

2023-08-22 17:15:04 -07:00

windowed_reductions.py

Remove the canonicalize_dtypes argument from mlir.ir_constant(s).

2023-08-17 06:44:12 -07:00