ROCm/jax

Pull Requests Commits

Fix lowering code for ROCm RNN

Ruturaj4 committed 255 days ago

34c2dbfd

[pallas] Fix spelling of 'fusible'.

chr1sj0nes committed 256 days ago

1926b99b

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

marksandler2 committed 256 days ago

0ed0fb7c

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

sharadmv committed 256 days ago

4fa3cd91

Merge pull request #27999 from vfdev-5:fix-skip-test-pattern-with-opt-deps

Google-ML-Automation committed 256 days ago

69d21c69

Deprecate the contents of jax.util.

hawkinsp committed 256 days ago

57e33bcb

Fixed the way to skip tests using optional python dependencies

vfdev-5 committed 256 days ago

7b4b2f47

Remove obsolete python key path registry.

IvyZX committed 256 days ago

ab600c3e

Fix warnings in array_interoperability_test.

hawkinsp committed 256 days ago

a88486ca

Merge pull request #28012 from emilyfertig:emilyaf-random-docs-line

Google-ML-Automation committed 256 days ago

a64e7dc0

Merge pull request #27919 from kaixih:enable_doc_scaled_dot_fix

Google-ML-Automation committed 256 days ago

19be20fc

Merge pull request #28020 from jakevdp:issue-template

Google-ML-Automation committed 256 days ago

8bae29d6

Update issue template with correct URL for untemplated issue

jakevdp committed 256 days ago

73305e03

Fix stablehlo version comparison in test utilities.

hawkinsp committed 256 days ago

8930a67e

Reinstate lifegiving chaos line to docs.

emilyfertig committed 256 days ago

1fcb2b47

Merge pull request #28007 from jakevdp:int-power

Google-ML-Automation committed 256 days ago

d0149126

Merge pull request #27966 from jakevdp:jit-signature

Google-ML-Automation committed 256 days ago

6fcb036b

Disable unknown warning option error on Mac.

nitins17 committed 256 days ago

785d0775

jnp.power: better docs for invalid input

jakevdp committed 256 days ago

42542fea

Merge pull request #27993 from gnecula:explain_timing

Google-ML-Automation committed 256 days ago

30669dc2

jax.jit: deprecate non-standard call signature.

jakevdp committed 256 days ago

ceca6ec1

Update XLA dependency to use revision

Google-ML-Automation committed 256 days ago

5af59257

Finalize deprecation of vectorized argument in callbacks.

dfm committed 256 days ago

1b1bd071

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

gnecula committed 257 days ago

077d134d

Merge pull request #27971 from ywrt:patch-1

Google-ML-Automation committed 257 days ago

b6c6c1c2

Merge pull request #27997 from apaszke:fix-tabs

Google-ML-Automation committed 257 days ago

9abc74d4

Remove accidental tab characters from Pallas:MGPU docs

apaszke committed 257 days ago

11a6abc6

Merge pull request #27925 from dfm:debug-rtds

Google-ML-Automation committed 257 days ago

95e0c2a1

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

gnecula committed 257 days ago

b8df4749

Merge pull request #27982 from apaszke:mgpu-wgmma-docs

Google-ML-Automation committed 257 days ago

ec8c0658

Older