ROCm/jax

Pull Requests Commits

[Mosaic TPU] Validate inserted layout in relayout-insertion pass.

bythew3i committed 1 year ago

9f842909

Merge pull request #25699 from yliu120:fix_iota

Google-ML-Automation committed 1 year ago

634b45bf

jnp.einsum: default to optimize='auto'

jakevdp committed 1 year ago

2f7204ff

bazel: export serialization.fbs for downstream usage

maxwillzq committed 1 year ago

c39e38fe

Merge pull request #25675 from jakevdp:dep-lpmn

Google-ML-Automation committed 1 year ago

74be8bd9

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

18b193cb

Deprecate scipy.special.lpmn & lpmn_values

jakevdp committed 1 year ago

245a13a3

Fixes tril/triu comments (they were flipped)

marksandler2 committed 1 year ago

6c87bf38

fix the degenerated case

yliu120 committed 1 year ago

3ff000ee

[shape_poly] Remove old non_negative support.

gnecula committed 1 year ago

e87a2a59

Disable `avxvnniint8` when building with Clang version < 19, or GCC < 13.

belitskiy committed 1 year ago

f2e210b3

Temporarily allow deprecation warnings for `scipy.special.lpmn` and `scipy.special.sph_harm`.

dfm committed 1 year ago

512d5450

Merge pull request #25715 from ROCm:ci_build_code_fixes-upstream

Google-ML-Automation committed 1 year ago

4de7794f

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

d0a92c5c

Add SMEM as a supported Pallas output memory space.

Google-ML-Automation committed 1 year ago

54fd738e

Merge branch 'jax-ml:main' into activation-offloading-doc

zhenying-liu committed 1 year ago

Verified 5e3a692d

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

9af29700

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

e4278f78

Merge pull request #25713 from jakevdp:debug-printoptions

Google-ML-Automation committed 1 year ago

0f4677bc

jax.debug.print: respect local np.printoptions

jakevdp committed 1 year ago

33060632

[Mosaic] NFC: Pull out vreg related functions to util.

WindQAQ committed 1 year ago

57b21541

Compute cost-analysis on only one HLO module.

zacmustin committed 1 year ago

df36c298

Merge pull request #25686 from Mikcl:docs/working-with-pytrees-formatting

Google-ML-Automation committed 1 year ago

800f903f

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

726950b8

Merge pull request #25710 from apaszke:mgpu_dialect_fix

Google-ML-Automation committed 1 year ago

68483b8e

Fix OSS build for the Mosaic GPU dialect

apaszke committed 1 year ago

64433435

[Mosaic:TPU][NFC] Clean up unused variable

tlongeri committed 1 year ago

ac817b48

Merge pull request #25706 from pearu:pearu/log10-large

Google-ML-Automation committed 1 year ago

82001ed5

Merge pull request #25661 from rdyro:tb-nightly-instructions

Google-ML-Automation committed 1 year ago

04a0fbec

Don't use x32 mode for pallas_test

apaszke committed 1 year ago

7c984c60

Newer Older