ROCm/jax

Pull Requests Commits

[ROCm]: CI scripts update to handle ROCm versions

Rahul Batra committed 2 years ago

d7c149e5

[ROCm]: Fix duplicate deps include

Rahul Batra committed 2 years ago

9f7dd8d1

[ROCm]: Update docker file to ROCm5.6

Rahul Batra committed 2 years ago

d5c86f74

[ROCm]: Add hipblaslt in Dockerfile

Rahul Batra committed 2 years ago

69d6d22f

Merge pull request #16870 from skye:version

a-googler committed 2 years ago

88a60b80

Slightly downgrade xla version to avoid PJRT C API incompat

skye committed 2 years ago

e132a0e5

Merge pull request #16869 from skye:version

a-googler committed 2 years ago

c75e85da

Work around pytype error.

hawkinsp committed 2 years ago

a480aa8d

Update WORKSPACE and setup.py in preparation for jax/jaxlib 0.4.14 release

skye committed 2 years ago

0b24b2ba

Merge pull request #16433 from JGameCreation:patch-1

a-googler committed 2 years ago

7c0ef866

Update flags to use the ABSL typed flag API.

hawkinsp committed 2 years ago

76cda0ae

Merge pull request #16865 from google:xla

a-googler committed 2 years ago

f35f226b

Update XLA version to fix build failure.

hawkinsp committed 2 years ago

0cd02534

Move _tpu_ext.cc to jaxlib/mlir/_mlir_libs and set RPATH correctly

skye committed 2 years ago

a03d6e66

Merge pull request #16852 from hawkinsp:builddeps

a-googler committed 2 years ago

bcddc504

Merge pull request #16844 from jakevdp:jnp-put

a-googler committed 2 years ago

5c39a0d3

Previously, using sparse.todense on a BCSR matrix with sparse.sparsify would raise `NotImplementedError: sparse rule for todense is not implemented`. By adding the sparse rule, it will resolve this issue.

a-googler committed 2 years ago

735637e3

Check `build` and `wheel` are installed before building `jaxlib`.

hawkinsp committed 2 years ago

3c4527b6

Merge pull request #16826 from mattjj:issue16805

a-googler committed 2 years ago

416814df

Merge pull request #16849 from skye:mosaic_test_build_fix

a-googler committed 2 years ago

9e692774

Add implementation of jnp.put

jakevdp committed 2 years ago

88c42da7

Make //jax:tpu_custom_call respect --//jax:build_jaxlib=false

skye committed 2 years ago

d0b65f2a

Merge pull request #16846 from gnecula:poly_dot

a-googler committed 2 years ago

1054fe5a

Enable building jaxlib w/ Mosaic

sharadmv committed 2 years ago

3baa6e7a

[shape_poly] Fix handling of dot_general with different lhs_dtype and rhs_dtype

gnecula committed 2 years ago

c9f9f28b

Merge pull request #16842 from jakevdp:dynamic-slice-unsigned

a-googler committed 2 years ago

f66d3cf0

lax.dynamic_slice: avoid negative index correction for unsigned indices

jakevdp committed 2 years ago

0dbda849

Merge pull request #16833 from gnecula:poly_v8_1

a-googler committed 2 years ago

def6190d

Merge pull request #16839 from jakevdp:fix-core-deprecation

a-googler committed 2 years ago

b1166169

fix deprecations in core.py

jakevdp committed 2 years ago

3b6b9884

Older