ROCm/jax

Pull Requests Commits

Merge pull request #24355 from gnecula:exp_fix_doc

Google-ML-Automation committed 1 year ago

36ec513a

Reverts 30ba7f37e027dfe80f9aaad6cd52d9387dba8612

WindQAQ committed 1 year ago

a01f187e

Google-ML-Automation committed 1 year ago

9027fb38

Support single-process AutoPGLE usage.

Google-ML-Automation committed 1 year ago

96d5542a

Merge pull request #24326 from rajasekharporeddy:testbranch1

Google-ML-Automation committed 1 year ago

2e5920db

[export] Fix github links in the export documentation

gnecula committed 1 year ago

9aa79bff

Better docs for jax.numpy:equal and not_equal

rajasekharporeddy committed 1 year ago

aaefa822

Merge pull request #24351 from jakevdp:sinc-doc

Google-ML-Automation committed 1 year ago

f332dd56

Merge pull request #24348 from jakevdp:fix-unused-imports

Google-ML-Automation committed 1 year ago

7d7b619e

Cleanup: fix unused imports & mark exported names

jakevdp committed 1 year ago

de3191fa

Merge pull request #24347 from jakevdp:ruff-rules

Google-ML-Automation committed 1 year ago

6a000559

CI: enable additional ruff formatting checks

jakevdp committed 1 year ago

e1f280c8

Improve docs for jnp.sinc

jakevdp committed 1 year ago

6ca7228d

Merge pull request #24343 from jakevdp:fix-pyi

Google-ML-Automation committed 1 year ago

4d6064be

[ROCm] Fix README.md to update AMD JAX installation instructions

Ruturaj4 committed 1 year ago

3c3b08df

Merge pull request #24323 from selamw1:tile_doc

Google-ML-Automation committed 1 year ago

ebac2e44

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

a5b31237

Merge pull request #24322 from andportnoy:aportnoy/skip-flash-attention-unless-sm90

Google-ML-Automation committed 1 year ago

1e2d4bc5

Merge pull request #24341 from phu0ngng:cuda_graph_ex

Google-ML-Automation committed 1 year ago

089e4aa9

Merge pull request #23831 from nouiz:doc_policies

Google-ML-Automation committed 1 year ago

ead1c05a

Fix aliases in jax.numpy type interface file.

jakevdp committed 1 year ago

b574d2ce

added cmdBuffer traits

phu0ngng committed 1 year ago

d4bbb4fd

Do not allow `None` axis names in meshes.

bartchr808 committed 1 year ago

801fe87d

rm CmdBuffer traits

phu0ngng committed 1 year ago

82113cd0

added cudaGraph traits + use register_ffi_target()

phu0ngng committed 1 year ago

f3775aa2

[pallas:mosaic_gpu] Added `FragmentedArray.to_layout`

superbobry committed 1 year ago

bb271aaf

[Pallas TPU] Add a better error message for rank 1 block mappings check.

pajarskas committed 1 year ago

1222b4a5

[pallas:mosaic_gpu] Added a few more operations necessary to port Flash Attention

superbobry committed 1 year ago

4c0d8282

Merge pull request #24312 from jakevdp:gather-doc

Google-ML-Automation committed 1 year ago

56eea2b5

Make committed a public property of jax.Array.

yashk2810 committed 1 year ago

66c6292e

Newer Older