ROCm/jax

Pull Requests Commits

[pallas] Fix spelling of 'fusible'.

chr1sj0nes committed 1 year ago

1926b99b

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

marksandler2 committed 1 year ago

0ed0fb7c

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

sharadmv committed 1 year ago

4fa3cd91

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

Google-ML-Automation committed 1 year ago

69d21c69

Deprecate the contents of jax.util.

hawkinsp committed 1 year ago

57e33bcb

Fixed the way to skip tests using optional python dependencies

vfdev-5 committed 1 year ago

7b4b2f47

Remove obsolete python key path registry.

IvyZX committed 1 year ago

ab600c3e

Fix warnings in array_interoperability_test.

hawkinsp committed 1 year ago

a88486ca

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

Google-ML-Automation committed 1 year ago

a64e7dc0

Merge pull request #27919 from kaixih:enable_doc_scaled_dot_fix

Google-ML-Automation committed 1 year ago

19be20fc

Merge pull request #28020 from jakevdp:issue-template

Google-ML-Automation committed 1 year ago

8bae29d6

Update issue template with correct URL for untemplated issue

jakevdp committed 1 year ago

73305e03

Fix stablehlo version comparison in test utilities.

hawkinsp committed 1 year ago

8930a67e

Reinstate lifegiving chaos line to docs.

emilyfertig committed 1 year ago

1fcb2b47

Merge pull request #28007 from jakevdp:int-power

Google-ML-Automation committed 1 year ago

d0149126

Merge pull request #27966 from jakevdp:jit-signature

Google-ML-Automation committed 1 year ago

6fcb036b

Disable unknown warning option error on Mac.

nitins17 committed 1 year ago

785d0775

jnp.power: better docs for invalid input

jakevdp committed 1 year ago

42542fea

Merge pull request #27993 from gnecula:explain_timing

Google-ML-Automation committed 1 year ago

30669dc2

jax.jit: deprecate non-standard call signature.

jakevdp committed 1 year ago

ceca6ec1

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

5af59257

Finalize deprecation of vectorized argument in callbacks.

dfm committed 1 year ago

1b1bd071

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

gnecula committed 1 year ago

077d134d

Merge pull request #27971 from ywrt:patch-1

Google-ML-Automation committed 1 year ago

b6c6c1c2

Merge pull request #27997 from apaszke:fix-tabs

Google-ML-Automation committed 1 year ago

9abc74d4

Remove accidental tab characters from Pallas:MGPU docs

apaszke committed 1 year ago

11a6abc6

Merge pull request #27925 from dfm:debug-rtds

Google-ML-Automation committed 1 year ago

95e0c2a1

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

gnecula committed 1 year ago

b8df4749

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

Google-ML-Automation committed 1 year ago

ec8c0658

Add Pallas:MGPU documentation for WGMMA

apaszke committed 1 year ago

3a7cec85

Older