ROCm/jax

Pull Requests Commits

[NFC] Rename `standard_insert_pbroadcast` to `standard_insert_pvary`

yashk2810 committed 253 days ago

89a388af

[Mosaic:TPU] Fix bug in `rotateVregRows`

tlongeri committed 253 days ago

e482f909

jax.random.bernoulli: use higher-resolution sampler

jakevdp committed 253 days ago

5dadfddc

Fix lowering code for ROCm RNN

Ruturaj4 committed 253 days ago

20991f9f

Add explicit_axes section to the doc

yashk2810 committed 253 days ago

48f042c7

Update XLA dependency to use revision

Google-ML-Automation committed 253 days ago

e2846994

Deprecate the remaining exports from jax.lib.xla_client.

hawkinsp committed 253 days ago

1806af7c

[CI] Propagate halt connection to tpu tests

MichaelHudgins committed 253 days ago

b1033cc3

[CI] Update GPU optional presubmit naming

MichaelHudgins committed 253 days ago

bb493ba6

[pallas] Fix spelling of 'fusible'.

chr1sj0nes committed 253 days ago

50a3e85e

Adds a debugging message to assert, otherwise the error is pretty cryptic.

marksandler2 committed 253 days ago

c453578a

[Pallas/Fuser] Add basic closed over consts support to pull_block_spec

sharadmv committed 253 days ago

ba70f636

Fixed the way to skip tests using optional python dependencies

vfdev-5 committed 253 days ago

a37c7227

Deprecate the contents of jax.util.

hawkinsp committed 253 days ago

07228c10

Remove obsolete python key path registry.

IvyZX committed 253 days ago

5a14a0e1

Fix warnings in array_interoperability_test.

hawkinsp committed 253 days ago

cd5c3522

Reinstate lifegiving chaos line to docs.

emilyfertig committed 253 days ago

1fa363d7

Don't use default quant config

kaixih committed 253 days ago

33177d86

Deprecation warning

kaixih committed 253 days ago

48679d0f

kaixih committed 253 days ago

796d4684

Keep old scale_matmul arg names

kaixih committed 253 days ago

9922f226

Update issue template with correct URL for untemplated issue

jakevdp committed 253 days ago

844ee391

Fix stablehlo version comparison in test utilities.

hawkinsp committed 253 days ago

f2bc0af0

jnp.power: better docs for invalid input

jakevdp committed 253 days ago

fcc11654

jax.jit: deprecate non-standard call signature.

jakevdp committed 253 days ago

90d62e94

Disable unknown warning option error on Mac.

nitins17 committed 253 days ago

2aaa0d6c

[explain_cache_miss] Add to explanations the duration of the missed function call

gnecula committed 253 days ago

e258df56

Update XLA dependency to use revision

Google-ML-Automation committed 253 days ago

d08a1396

Finalize deprecation of vectorized argument in callbacks.

dfm committed 253 days ago

bbdf651c

Adjust test expectations for the tracing-cache-miss-explanations

gnecula committed 253 days ago

00c05e82

Newer Older