ROCm/jax

Pull Requests Commits

Add missing version guard in GPU tests for jnp.poly.

dfm committed 1 year ago

3556a833

Merge pull request #24968 from nireekshak:testingbranch

Google-ML-Automation committed 1 year ago

6929a97c

Merge pull request #24942 from jeertmans:patch-1

Google-ML-Automation committed 1 year ago

9d3eda17

fix(docs): typos in macro name

jeertmans committed 1 year ago

Verified d912034c

nireekshak committed 1 year ago

1458d3dd

[AutoPGLE] Use compile options to override debug options instead of XLA_FLAGS.

Google-ML-Automation committed 1 year ago

da50ad7e

Implement lax.pad in Pallas.

Google-ML-Automation committed 1 year ago

d397dd96

Merge pull request #24853 from yliu120:check_proxy_envs

Google-ML-Automation committed 1 year ago

12a43f1f

[Mosaic TPU] Support relayout for mask vector

bythew3i committed 1 year ago

0fe77bc9

Merge pull request #24861 from yliu120:add_versions

Google-ML-Automation committed 1 year ago

58103e5a

Merge pull request #24945 from hawkinsp:gamma

Google-ML-Automation committed 1 year ago

4a9346e4

Update jax.scipy.special.gamma and gammasgn to return NaN for negative integer inputs.

hawkinsp committed 1 year ago

c5e8ae80

[pallas] Minor simplifications to Pallas interpreter.

chr1sj0nes committed 1 year ago

45c9c0a5

Fix a bug where mesh checking was not correct

yashk2810 committed 1 year ago

2c68569a

Delete _normalized_spec from NamedSharding

yashk2810 committed 1 year ago

e904c177

Merge pull request #24958 from barnesjoseph:add-font-fallback

Google-ML-Automation committed 1 year ago

6952ddf4

Adds font fallbacks

barnesjoseph committed 1 year ago

d4316b57

Merge pull request #23585 from apivovarov:float8_e4m3

Google-ML-Automation committed 1 year ago

91891cb6

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

b3ca6c47

Merge pull request #24957 from hawkinsp:arm

Google-ML-Automation committed 1 year ago

16ed283f

[Pallas] Increase test coverage of pl.dot.

WindQAQ committed 1 year ago

a60ef6e9

Filter custom dtypes by supported_dtypes in `_LazyDtypes`.

hawkinsp committed 1 year ago

f3250516

Disable some complex function accuracy tests that fail on Mac ARM.

hawkinsp committed 1 year ago

461a2507

[SDY] fix JAX layouts tests for Shardy.

Varcho committed 1 year ago

0ed6eaeb

Merge pull request #24952 from jakevdp:fix-pyi

Google-ML-Automation committed 1 year ago

70b05f6c

fix typo in numpy/__init__.pyi

jakevdp committed 1 year ago

5bebd0f6

Return SingleDeviceSharding instead of GSPMDShardings when there is only 1 device during `compiled.input_shardings` call.

yashk2810 committed 1 year ago

6fe7b171

Merge pull request #24903 from jakevdp:logsumexp

Google-ML-Automation committed 1 year ago

297a4e5e

Make logaddexp and logaddexp2 into ufuncs

jakevdp committed 1 year ago

e9864c69

Merge pull request #24916 from jakevdp:update-lp

Google-ML-Automation committed 1 year ago

2de40e7d

Newer Older