6f50677

hilbert transform

joglekara Mar 21, 2023

2bdf6c9

newline

joglekara Mar 21, 2023

cf5ddb0

dtype errors

joglekara Mar 24, 2023

e120f13

passing tests

joglekara Mar 24, 2023

0fd7920

linters

joglekara Mar 24, 2023

abf8947

Optional int

joglekara Mar 24, 2023

cbc25dc

Raise a better error message when there is a device assignment mismat…

yashk2810 Mar 21, 2023

ab91657

Fix inspect_array_sharding with grad.

nouiz Mar 21, 2023

c092312

Relax type annotations on lax slicing functions.

hawkinsp Mar 21, 2023

8e2e2f5

[jax2tf] Clean up the jax2tf sharding_tests.

gnecula Mar 21, 2023

41c1d93

Remove the config.jax_array and jax_jit_pjit_api_merge flag usage sin…

yashk2810 Mar 21, 2023

8717494

Document ShapeDtypeStruct

jakevdp Mar 21, 2023

fa19118

[jax2tf] Add back_compat test for LuDecomposition

atondwal Mar 21, 2023

ddab64a

LuDecomposition moved from fallback path to custom_call

atondwal Mar 21, 2023

48db6c8

[PJRT C API] Add parsing PJRT client create options from json file.

Mar 21, 2023

ffc8a34

`custom_vjp` symbolic zeros support

froystig Mar 14, 2023

ce3f534

[jax2tf] Fix grad of pjit in native lowering.

gnecula Mar 21, 2023

4aa8ae9

Fix mypy failures in jax2tf.

hawkinsp Mar 22, 2023

f4a40dc

add wald random generator

JiaYaobo Mar 21, 2023

92e79b3

Revert: `custom_vjp` symbolic zeros support

hawkinsp Mar 22, 2023

fcac7b4

jax.random: remove scale from wald function

jakevdp Mar 22, 2023

78488f0

Improve handling of dynamic shapes in jax native serialization

GleasonK Mar 22, 2023

c5ba4d3

make mlir arg and result names work with pmap

mattjj Mar 18, 2023

499372d

DOC: add formulae for distributions in jax.random

jakevdp Mar 22, 2023

aa46778

WAR the dependency issue in the nightly CI container.

nouiz Mar 22, 2023

f9d73cb

Add print statement to help debug spurious test failure

sharadmv Mar 22, 2023

3039951

add experimental jax_log_checkpoint_residuals option

mattjj Mar 22, 2023

4a27af3

Redefine `compile_and_serialize` as `serialize(lowered.compile())`.

pschuh Mar 23, 2023

6fee63c

enable pjit recursive typechecking

mattjj Mar 22, 2023

dd9f178

fix typo: "one of more" -> "one or more"

a-googler Mar 23, 2023

b0e0a94

Fix isinstance(k, PRNGKeyArray) on PRNGKeyArray subclasses

Conchylicultor Mar 23, 2023

a9b4310

[jax2tf] Fix tests broken by upgrade of XlaCallModule

gnecula Mar 23, 2023

8f6e3c4

[XLA:Python] Change JAX and the XLA Python extension to get NumPy bfl…

hawkinsp Mar 23, 2023

c68c3d3

jnp.mean: fix incorrect return value for large arrays

jakevdp Mar 21, 2023

75fcc3a

[XLA:Python] Allow passing ExecutableBuildOptions to outfeed receiver.

hawkinsp Mar 23, 2023

0e2cf94

Add missing file

nouiz Mar 23, 2023

8f4b8a0

Fix loc and scale parameters in scipy.logistic. Add CDF and SF for se…

b0nce Mar 12, 2023

5b30f8e

Move jax._src.typing into a separate Bazel target.

hawkinsp Mar 23, 2023

7a326b3

Deprecated xla_call_p since it has been replaced with pjit.pjit_p

yashk2810 Mar 23, 2023

6ee6598

Fix mypy issue in jax/experimental/jet.py

jakevdp Mar 23, 2023

180f12e

add trailing-whitespace pre-commit hook

cgarciae Mar 23, 2023

fdea9e6

[jax2tf] Minor addition to the documentation

gnecula Mar 23, 2023

1e356cf

internal: refactor array methods into separate private submodule

jakevdp Mar 23, 2023

23b0743

Refactor special functions into their own module.

atondwal Mar 23, 2023

071d9b9

improve scan error messages

mattjj Mar 23, 2023

175cd37

[jax2tf] Create a jax_export library with JAX-only pieces for native …

gnecula Mar 23, 2023

4994472

Add padding option "SAME_LOWER" for ticket https://github.com/google/…

maxwillzq Mar 23, 2023

32b8c42

[jax2tf] A simple failing test on TPU with native serialization

gnecula Mar 23, 2023

195f847

Migrate igamma_p off xla_fallback

atondwal Mar 23, 2023

f63a09c

lax_numpy: move quantile-based functions to reductions.py

jakevdp Mar 23, 2023

d138853

Increase minimum NumPy version to 1.21.

hawkinsp Feb 6, 2023

b6b1e42

Remove PJRT C API bypass.

skye Mar 24, 2023

fd36ed6

[dynamic-shapes] make dynamic shape staging-to-jaxpr work with pjit

mattjj Mar 23, 2023

10aeadb

fix jax.Array.round()

mattjj Mar 24, 2023

9d60043

[jaxlib] fix build w/ depenency on stablehlo_serialization

atondwal Mar 24, 2023

a88509a

Migrate igammac_p off xla_fallback path

atondwal Mar 24, 2023

c6bc3ed

Migrate igamma_grad_a_p off xla_fallback

atondwal Mar 24, 2023

aa37c74

Migrate random_gamma_grad off xla_fallback

atondwal Mar 24, 2023

2900c78

remove another dependency not currently needed.

nouiz Mar 24, 2023

f1af74f

WAR ssh timeout like:

nouiz Mar 23, 2023

14f1f60

After the SPMD bug fix, always take the _rewriting_take route for get…

yashk2810 Mar 24, 2023

b9bd60d

Guard ArrayImpl checks by xla_extension_version.

pschuh Mar 24, 2023

5c08753

[1/n] store embedded tf.graph to stablehlo.custom_call

maxwillzq Mar 24, 2023

5b43258

Delete remote TPU support.

hawkinsp Mar 24, 2023

f83a347

docs

joglekara Mar 24, 2023

3b11662

Merge branch 'google:main' into hilbert

joglekara Mar 24, 2023