ROCm/jax

Pull Requests Commits

Merge pull request #23928 from nvcastet:fix_mosaic_barrier

Google-ML-Automation committed 1 year ago

a6b4648e

Improve documentation for jax.jacobian

jakevdp committed 1 year ago

cf51ee7e

[Pallas/Mosaic GPU] Replace tiling/transpose fields of GPUBlockSpec with a transform list

apaszke committed 1 year ago

3c25da2c

[pallas::mosaic_gpu] Turn the accumulator into a reference

cperivol committed 1 year ago

b6d668e0

Merge pull request #23899 from rajasekharporeddy:testbranch2

Google-ML-Automation committed 1 year ago

f6fdfb45

Better doc for jnp.trace

rajasekharporeddy committed 1 year ago

8ffeb238

Merge pull request #23924 from jburnim:jburnim_doc_typo_fix

Google-ML-Automation committed 1 year ago

911acf1b

Roll back cl/678765762 [Mosaic TPU] Support bitcast without forcing retiling.

bythew3i committed 1 year ago

e4ca4f5a

Change `CLANG_CUDA_COMPILER_PATH` set order. Add `--config=cuda_clang` to build.py

Google-ML-Automation committed 1 year ago

f1b3251b

Improve docs for jnp.pad

jakevdp committed 1 year ago

ad6c3a7f

[Mosaic GPU] Fix mbarrier inline ptx for newer CTK

nvcastet committed 1 year ago

08629a42

Fix make_remote_async_copy -> make_async_remote_copy in async doc.

jburnim committed 1 year ago

a1f2edc9

Remove `CC="/usr/lib/llvm-18/bin/clang"` from clang config in .bazelrc

Google-ML-Automation committed 1 year ago

ce99c18a

Merge pull request #23916 from jakevdp:interp-doc

Google-ML-Automation committed 1 year ago

5d4cae07

[Pallas] Add scalar f32 downcast test cases.

Google-ML-Automation committed 1 year ago

70346bda

Finalize deprecation of pretty-printing utils in `jax.core.pp_*`

Jake VanderPlas committed 1 year ago

e05c37c6

[Mosaic TPU] Support bitcast without forcing retiling.

bythew3i committed 1 year ago

37641dd4

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

c93b272b

Improve documentation for jnp.interp

jakevdp committed 1 year ago

ee6fd5ae

Merge pull request #23917 from dfm:gh23895

Google-ML-Automation committed 1 year ago

0f84c2c6

Reverts eff00cc4499cfe3f3f24bafda6c1ecf908232ff3

tomnatan30 committed 1 year ago

6cf09f8c

Merge pull request #23874 from rajasekharporeddy:testbranch1

Google-ML-Automation committed 1 year ago

a20f79d8

Merge pull request #23914 from rajasekharporeddy:testbranch3

Google-ML-Automation committed 1 year ago

f126705d

Merge pull request #23382 from jax-ml:dependabot/github_actions/actions/setup-python-5.2.0

Google-ML-Automation committed 1 year ago

8806e0b6

Merge pull request #23910 from 8bitmp3:patch-3

Google-ML-Automation committed 1 year ago

8edc0fca

Reverts dffac29e63de6a51047fe77cf9d553ab762ef19b

hawkinsp committed 1 year ago

111f13e2

Fix dtype bug in jax.scipy.fft.idct

dfm committed 1 year ago

96268dca

Fixed `pl.debug_print`ing of scalar fragmented arrays under Mosaic GPU

superbobry committed 1 year ago

b49d8b26

rajasekharporeddy committed 1 year ago

13774d13

Increase sharding of checkify_test on TPU to fix CI flakes.

hawkinsp committed 1 year ago

19494137

Newer Older