ROCm/jax

Pull Requests Commits

Merge pull request #21077 from merrymercy:patch-1

a-googler committed 1 year ago

500da57e

Start jax and jaxlib 0.4.27 release

yashk2810 committed 1 year ago

70b44772

[Mosaic GPU] Adjust memref.expand_shape construction to pass in the new args

apaszke committed 1 year ago

326adc01

Update XLA dependency to use revision

a-googler committed 1 year ago

3e5a18f1

Merge pull request #21081 from hawkinsp:sourcemap

a-googler committed 1 year ago

cb0c4985

Fix that the insufficient output HBM buffer init would cause the <unk> token generated for quantized int8 model.

a-googler committed 1 year ago

4de34648

Merge pull request #21070 from shuhand0:rel0.0.7

a-googler committed 1 year ago

eee2783e

Merge pull request #20327 from selamw1:add_examples

a-googler committed 1 year ago

f6d88525

fix jaxlib config name

shuhand0 committed 1 year ago

aac36799

improve documentation for ix_

selamw1 committed 1 year ago

9caf59d6

Merge pull request #20842 from Micky774:array-api-default-promotion

a-googler committed 1 year ago

3d3cb0bd

Merge pull request #21079 from jakevdp:tensorinv

a-googler committed 1 year ago

1013b1ad

Merge pull request #21087 from jakevdp:upstream-print-version

a-googler committed 1 year ago

bb6aa12f

Compute source maps when pretty-printing jaxprs.

hawkinsp committed 1 year ago

d014f5dc

CI: print numpy/scipy version in upstream job

jakevdp committed 1 year ago

7d96e78a

Add an experimental, Clang version of the Windows CI job.

a-googler committed 1 year ago

a265e421

jnp.linalg tensorinv & tensorsolve: improve implementation & docs

jakevdp committed 1 year ago

4a363156

Merge pull request #21084 from jakevdp:fix-upstream

a-googler committed 1 year ago

d26bd735

Merge pull request #21078 from jakevdp:numpy-linalg-doc

a-googler committed 1 year ago

7e9ef1e4

Add a config for using Clang on Windows.

a-googler committed 1 year ago

fb65ba4a

Pass `bazel_options` directly to the Bazel command, instead of into .bazelrc.

a-googler committed 1 year ago

8ba5c647

jnp.linalg: improve API documentation

jakevdp committed 1 year ago

40b2d485

random_lax_test: fix kstest for newer NumPy

jakevdp committed 1 year ago

6f7ebff5

Refactored common upcast for integral-type accumulators

Micky774 committed 1 year ago

34c5163f

Fix a typo in jax.jit docstring

merrymercy committed 1 year ago

Verified 0eed28a0

Don't create temp directory when module is getting imported.

a-googler committed 1 year ago

76814937

Update XLA dependency to use revision

a-googler committed 1 year ago

047ea210

Update XLA dependency to use revision

a-googler committed 1 year ago

a1c82219

Merge pull request #21056 from mattjj:vmap-grad-remat-shmap-bug

a-googler committed 1 year ago

1b804a77

[shard_map] better fix for spmd_axis_name issues with shmap residuals

mattjj committed 1 year ago

7a87010f

Older