ROCm/jax

Pull Requests Commits

[ROCm] Fix invalid repo url for EL path

mrodden committed 1 year ago

83f88b3c

Fix a bug where treedef.flatten_up_to(...) was overly permissive for None treedefs.

hawkinsp committed 1 year ago

255c3030

Fixed a static type error in Mosaic GPU lowering

superbobry committed 1 year ago

5b9c73f3

Merge pull request #23593 from rajasekharporeddy:testbranch2

Google-ML-Automation committed 1 year ago

f291fb9f

Merge pull request #23603 from hawkinsp:jaxbuild2

Google-ML-Automation committed 1 year ago

245c2b00

Redisable one more complex arctan test.

hawkinsp committed 1 year ago

42a295c7

Merge pull request #23602 from hawkinsp:arctan

Google-ML-Automation committed 1 year ago

c2b1a71b

Redisable complex arctan tests on older jaxlibs.

hawkinsp committed 1 year ago

c2adbf9e

Lower the minimum jaxlib version to 0.4.31, since 0.4.32 was yanked from pypi.

hawkinsp committed 1 year ago

46a7be6d

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

39c39acd

Merge pull request #23586 from rajasekharporeddy:testbranch3

Google-ML-Automation committed 1 year ago

522ad799

Delete jax.xla_computation since it's been 3 months since it was deprecated.

yashk2810 committed 1 year ago

de9b98e0

Merge pull request #23591 from jaro-sevcik:gpu-enable-auto-layout-tests

Google-ML-Automation committed 1 year ago

985e74f2

jaro-sevcik committed 1 year ago

20811648

Refactor gpusolver kernel definitions into separate build target.

dfm committed 1 year ago

a3bf75e4

Merge pull request #23587 from pearu:pearu/update-atan-atanh-tests

Google-ML-Automation committed 1 year ago

2067da81

Improve docs for jax.numpy: diff and ediff1d

rajasekharporeddy committed 1 year ago

191d7ccd

Improve doc for jnp.resize

rajasekharporeddy committed 1 year ago

5234173f

Merge pull request #23534 from ayaka14732:pallas-indexing-3

Google-ML-Automation committed 1 year ago

baf9cc70

Update complex arctan and arctanh accuracy tests

pearu committed 1 year ago

3c024294

Merge pull request #23572 from rajasekharporeddy:testbranch2

Google-ML-Automation committed 1 year ago

0faac82d

Flip jax_pmap_no_rank_reduction by default to True.

pschuh committed 1 year ago

bf2237a1

Enable the enhanced TPU launch barrier on all TPU generations.

hawkinsp committed 1 year ago

e64de982

Disable complex arctan test in mnegj.real part of plane.

hawkinsp committed 1 year ago

9a8dc6b4

Merge pull request #23578 from hawkinsp:postrelease

Google-ML-Automation committed 1 year ago

e86d77c4

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

7ce8ff29

Add link to XLA documentation for building JAX with CUDA from sources.

Google-ML-Automation committed 1 year ago

7c8508e5

Update version numbers after v0.4.32 release.

hawkinsp committed 1 year ago

3e81ae53

Prepare for v0.4.32 release.

hawkinsp committed 1 year ago

1594d2f3

Merge pull request #22702 from keshavb96:rm_custom_partitioning_pointer

Google-ML-Automation committed 1 year ago

859188b3

Older