ROCm/jax

Pull Requests Commits

Merge pull request #63 from ROCm/fix-scipy-missing-tril

mrodden committed 1 year ago

Verified a2fba1f0

Pin scipy to 1.12.0 for jax 0.4.23 builds

mrodden committed 1 year ago

Verified ffc394bb

Implemented CI Build support for JAX whl and Docker

idass1990 committed 2 years ago

40eef0e2

[ROCm]: CI Updates

Rahul Batra committed 2 years ago

73cc33b0

[ROCm]: Disable sparse_bcoo_bcsr_test.py::test_bcsr_dot_general for complex types

Rahul Batra committed 2 years ago

86e1ab37

[ROCm]: Updates hip headers path for ROCm 6.0

Rahul Batra committed 2 years ago

be3f3c45

[ROCm]: Disable sparse bcsr test

Rahul Batra committed 2 years ago

721a0753

[ROCm]: Disable some interpretor test for float16

Rahul Batra committed 2 years ago

e32bc894

[ROCm]: Disable test for rocm in tests_shape_poly_test

Rahul Batra committed 2 years ago

32c7d357

[ROCm]: Disable multi-platform tests in export_test

Rahul Batra committed 2 years ago

b1c7fc32

[ROCm]: Disable export_harnesses_multi_platform_test on ROCm

Rahul Batra committed 2 years ago

e4743361

[ROCm]: Comment out GPU_LOWERING_ENABLED in sparse_test

Rahul Batra committed 2 years ago

c9f4b37f

[ROCm]: Disable sparse tests

invalid-email-address committed 2 years ago

40c6c322

Disable linalg_test::EighTinyNorm

Rahul Batra committed 2 years ago

bb0e6211

[ROCm]: Dockerfile updates

Rahul Batra committed 2 years ago

7f542aa1

[ROCm]: Lower sparse(some) ops correctly for ROCm

Rahul Batra committed 2 years ago

ebe7fb17

Prepare for emergency jax and jaxlib 0.4.23 release

yashk2810 committed 2 years ago

0017b3a6

Add patch to suppress XLA:GPU logging.

hawkinsp committed 2 years ago

432cd182

Finish jax and jaxlib 0.4.22 release

yashk2810 committed 2 years ago

cccb5650

split the random generalized normal test and skip its K-S half

froystig committed 2 years ago

a85dd62c

Update thinking-in-jax working-with-pytrees

8bitmp3 committed 2 years ago

7edc80d6

Merge pull request #18581 from 8bitmp3:jax-docs-thinking-in-jax

a-googler committed 2 years ago

29ed3cd4

Reverts 3c07c10a9a55f9a32390dd10cf3f420bdf3f1ed8

yashk2810 committed 2 years ago

b52bcc16

Upgrade How to think in JAX

8bitmp3 committed 2 years ago

a4003bf0

Fix the pjit flakey test. The test was weirdly written in the first place. The current suspicion is that `x.copy()` inside jit made it flakey. Also delete some duplicate tests from the time of migrating to jax.Array

yashk2810 committed 2 years ago

e8888065

Start jax and jaxlib 0.4.22 release

yashk2810 committed 2 years ago

2e633522

Remove the `jax_require_devices_during_lowering flag since it was temporary. Added the semi-breaking change to Changelog.md.

yashk2810 committed 2 years ago

3c07c10a

Merge pull request #18869 from 8bitmp3:small-fix-jax-pytrees

a-googler committed 2 years ago

0281596e

Merge pull request #18582 from 8bitmp3:jax-docs-debugging-101

a-googler committed 2 years ago

d269c581

Merge pull request #18961 from mattjj:issue18955

a-googler committed 2 years ago

4459991d

Older