ROCm/jax

Pull Requests Commits

Remove obsolete python key path registry.

IvyZX committed 259 days ago

ab600c3e

Fix warnings in array_interoperability_test.

hawkinsp committed 259 days ago

a88486ca

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

Google-ML-Automation committed 259 days ago

a64e7dc0

Merge pull request #27919 from kaixih:enable_doc_scaled_dot_fix

Google-ML-Automation committed 259 days ago

19be20fc

Merge pull request #28020 from jakevdp:issue-template

Google-ML-Automation committed 259 days ago

8bae29d6

Update issue template with correct URL for untemplated issue

jakevdp committed 259 days ago

73305e03

Fix stablehlo version comparison in test utilities.

hawkinsp committed 259 days ago

8930a67e

Reinstate lifegiving chaos line to docs.

emilyfertig committed 259 days ago

1fcb2b47

Merge pull request #28007 from jakevdp:int-power

Google-ML-Automation committed 259 days ago

d0149126

Merge pull request #27966 from jakevdp:jit-signature

Google-ML-Automation committed 259 days ago

6fcb036b

Disable unknown warning option error on Mac.

nitins17 committed 259 days ago

785d0775

jnp.power: better docs for invalid input

jakevdp committed 259 days ago

42542fea

Merge pull request #27993 from gnecula:explain_timing

Google-ML-Automation committed 259 days ago

30669dc2

jax.jit: deprecate non-standard call signature.

jakevdp committed 259 days ago

ceca6ec1

Update XLA dependency to use revision

Google-ML-Automation committed 259 days ago

5af59257

Finalize deprecation of vectorized argument in callbacks.

dfm committed 259 days ago

1b1bd071

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

gnecula committed 259 days ago

077d134d

Merge pull request #27971 from ywrt:patch-1

Google-ML-Automation committed 259 days ago

b6c6c1c2

Merge pull request #27997 from apaszke:fix-tabs

Google-ML-Automation committed 259 days ago

9abc74d4

Remove accidental tab characters from Pallas:MGPU docs

apaszke committed 259 days ago

11a6abc6

Merge pull request #27925 from dfm:debug-rtds

Google-ML-Automation committed 259 days ago

95e0c2a1

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

gnecula committed 259 days ago

b8df4749

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

Google-ML-Automation committed 259 days ago

ec8c0658

Add Pallas:MGPU documentation for WGMMA

apaszke committed 259 days ago

3a7cec85

Merge pull request #27980 from gnecula:tracing_cache

Google-ML-Automation committed 260 days ago

6ca623f7

Merge pull request #27973 from mattjj:iadd-gotcha

Google-ML-Automation committed 260 days ago

1af747b3

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

mattjj committed 260 days ago

13c7183c

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

tlongeri committed 260 days ago

2e4c0ec7

Minor improvements to doc for jax.nn.logsumexp.

carlosgmartin committed 260 days ago

2336cd16

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

gnecula committed 260 days ago

f070cdec

Newer Older