ROCm/jax

Pull Requests Commits

[ROCm]: fix tsl path

Ruturaj4 committed 2 years ago

97bf2d2b

Merge pull request #20571 from hawkinsp:release

a-googler committed 2 years ago

fed7efd7

Prepare for 0.4.26 release.

hawkinsp committed 2 years ago

61493263

Avoid unnecessary fori_loop when calculating the block indices.

voutcn committed 2 years ago

9bb3f79b

Merge pull request #20562 from olupton:bind-to-all

a-googler committed 2 years ago

85cb1693

Merge pull request #20560 from pearu:pearu/log1p-fixes

a-googler committed 2 years ago

57ee6b75

Use scipy 1.13.0 instead of scipy 1.13.0rc1 in CI.

hawkinsp committed 2 years ago

1db4af1c

Workaround mpmath bug (mpmath/mpmath#774) in log1p at complex infinities

pearu committed 2 years ago

9a7fb898

Fix tests that fail if enable_checks is true under NumPy 2.0.0rc1.

hawkinsp committed 2 years ago

e2f47748

jax.distributed.initialize: specify bind address.

olupton committed 2 years ago

2dd1b3d6

Merge pull request #20534 from gnecula:callback_64bit

a-googler committed 2 years ago

d89f0d66

[callback] Allow external callbacks to return 64-bit values in 32-bit mode

gnecula committed 2 years ago

35b1cb79

Update XLA dependency to use revision

a-googler committed 2 years ago

dcd45c8d

[JAX] Update JAX CI dockerfiles to use NumPy 2.0.0rc1, SciPy 1.13.0rc1, and ml_dtypes 0.4.0.

hawkinsp committed 2 years ago

1159691f

Disable a random distribution test that appears to fail under scipy 1.13.0rc1.

hawkinsp committed 2 years ago

b7401872

Merge pull request #20552 from jakevdp:geomspace-complex

a-googler committed 2 years ago

6b582f59

Re-enable persistent cache on cpu.

a-googler committed 2 years ago

88dd29a0

[PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h.

hawkinsp committed 2 years ago

1baed9b2

[Pallas TPU] Relax windowing restriction when lowering mapped grids

sharadmv committed 2 years ago

318ae893

jnp.geomspace: make complex behavior consistent with NumPy 2.0

jakevdp committed 2 years ago

fd7c85b3

Removed unnecessary BUILD dependencies from :ops_test

superbobry committed 2 years ago

f74f4ed4

Merge pull request #20548 from jakevdp:uint-floordiv

a-googler committed 2 years ago

a54eb81d

Merge pull request #20536 from jakevdp:broadcast-to

a-googler committed 2 years ago

e282bf57

jnp.floor_div: lower directly to div for unsigned int

jakevdp committed 2 years ago

e99a3051

Merge pull request #20542 from jakevdp:solve-test

a-googler committed 2 years ago

3c808123

Fix typo in Pallas design

sharadmv committed 2 years ago

87aee90e

test: remove solve test case that is invalid under NumPy 2.0

jakevdp committed 2 years ago

37a34bf6

Merge pull request #20543 from jakevdp:skip-geomspace

a-googler committed 2 years ago

87b869dd

test: skip complex geomspace test under numpy 2.0

jakevdp committed 2 years ago

c79f54f7

This CL introduces 'PluginProgram' in IFRT and exposes this in python via `xla_client.compile_ifrt_program()`.

a-googler committed 2 years ago

fb55d591

Older