ROCm/jax

Pull Requests Commits

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

Merge pull request #27980 from gnecula:tracing_cache

Google-ML-Automation committed 1 year ago

6ca623f7

Merge pull request #27973 from mattjj:iadd-gotcha

Google-ML-Automation committed 1 year ago

1af747b3

add a brief description of the jax.Array-has-no-__iadd__ gotcha

mattjj committed 1 year ago

13c7183c

[Mosaic:TPU] Add some invariant checking in VectorLayout ctor

tlongeri committed 1 year ago

2e4c0ec7

Minor improvements to doc for jax.nn.logsumexp.

carlosgmartin committed 1 year ago

2336cd16

[explain-cache-miss] Improve tracing-cache-miss explanations

gnecula committed 1 year ago

f070cdec

Newer Older